sciai.architecture.Normalize

查看源文件
class sciai.architecture.Normalize(lb, ub)[源代码]

使用指定的下界和上界对输入进行归一化。

参数:
  • lb (Tensor) - 下界。

  • ub (Tensor) - 上界。

输入:
  • inputs (Tensor) - 要归一化的输入Tensor。

输出:

Tensor,归一化投影到 [-1, 1]。

支持平台:

GPU CPU Ascend

样例:

>>> import mindspore as ms
>>> from mindspore import ops, nn
>>> from sciai.architecture.basic_block import Normalize
>>> x = ops.ones((3, 2), ms.float32)
>>> lb, ub = ops.Tensor([0, -0.5], ms.float32), ops.Tensor([2, 3.5], ms.float32)
>>> normalize = Normalize(lb, ub)
>>> res = normalize(x)
>>> print(res)
[[ 0.   -0.25]
[ 0.   -0.25]
[ 0.   -0.25]]