mirror of
https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git
synced 2025-06-11 02:26:00 +00:00
12 lines
271 B
Python
12 lines
271 B
Python
import numpy as np
|
|
|
|
from numpy.typing import ArrayLike
|
|
from typing import Any, Tuple
|
|
|
|
|
|
class Transform:
|
|
def apply(self, t: float, x: ArrayLike):
|
|
y = self(t, x)
|
|
assert x.shape == y.shape, "Transform has to maintain the array's shape."
|
|
|
|
return y |