ALPAKA/SPATZ
1
0
Fork 0
mirror of https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git synced 2025-09-29 05:17:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Current state of the simulation

Browse Source
This commit is contained in:
dario
2024-04-04 15:22:50 +02:00
parent a95252a361
commit 8b80a6d9d5
4 changed files with 49 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
spatz/transforms/noise.py
View File

@@ -13,15 +13,25 @@ class GaussianNoise(Transform):
self.__mu = mu
self.__sigma = sigma
def __call__(self, t: float, x: ArrayLike) -> ArrayLike:
assert np.shape(self.__mu) == np.shape(x), "Mu and x have to match in shape."
def __call__(self, _: float, x: ArrayLike) -> ArrayLike:
if np.isscalar(x):
noise = np.random.normal(0, 1)
x += self.__sigma * noise + self.__mu
else:
dim = len(x)
if np.isscalar(self.__sigma):
sigma = np.identity(dim) * self.__sigma
else:
sigma = self.__sigma
if np.isscalar(self.__mu):
mu = np.ones(dim)
else:
mu = self.__mu
noise = np.random.normal(0, 1, np.shape(x))
x += self.__sigma @ noise + self.__mu
x += sigma @ noise + mu
return x