diff --git a/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py b/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py index 29de110169d55a715495615ad67cd2583b8bb2da..6867cdd2393482c6a996e8512b9ddf966fc3f8f7 100644 --- a/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py +++ b/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py @@ -21,8 +21,8 @@ from scipy.interpolate import interp2d import matplotlib import matplotlib.pyplot as plt import matplotlib.animation as animation -from mindspore import jit_class, Tensor -from mindflow.loss import RelativeRMSELoss +from mindspore import jit_class, ops +from mindflow.core import RelativeRMSELoss from mindflow.utils import print_log from mindflow.cell import poly_data @@ -33,8 +33,8 @@ class UnitGaussianNormalizer(): def __init__(self, x, eps=0.00001): super(UnitGaussianNormalizer, self).__init__() - self.mean = Tensor.mean(x) - self.std = Tensor.std(x) + self.mean = ops.mean(x) + self.std = ops.std(x) self.eps = eps def encode(self, x):