mirror of
https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git
synced 2025-09-28 21:17:33 +00:00
SPATZ migration + proper directory structure
This commit is contained in:
12
spatz/transforms/transform.py
Normal file
12
spatz/transforms/transform.py
Normal file
@@ -0,0 +1,12 @@
|
||||
import numpy as np
|
||||
|
||||
from numpy.typing import ArrayLike
|
||||
from typing import Any, Tuple
|
||||
|
||||
|
||||
class Transform:
|
||||
def apply(self, t: float, x: np.array):
|
||||
y = self(t, x)
|
||||
assert x.shape == y.shape, "Transform has to maintain the array's shape."
|
||||
|
||||
return y
|
Reference in New Issue
Block a user