mindflow.cell.InputScale

class mindflow.cell.InputScale(input_scale, input_center=None)[源代码]

将输入值缩放到指定的区域。按照 \((x_i - input_center)*input_scale\) 变换。

参数:
  • input_scale (list) - 输入的比例因子。

  • input_center (Union[list, None]) - 坐标转换的中心位置,理解为偏移量。默认值: None

输入:
  • input (Tensor) - shape为 \((*, channels)\) 的Tensor。

输出:

shape为 \((*, channels)\) 的Tensor。

异常:
  • TypeError - 如果 input_scale 不是list类型。

  • TypeError - 如果 input_center 不是list类型或者 None

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> from mindflow.cell import InputScale
>>> from mindspore import Tensor
>>> inputs = np.random.uniform(size=(16, 3)) + 3.0
>>> inputs = Tensor(inputs.astype(np.float32))
>>> input_scale = [1.0, 2.0, 4.0]
>>> input_center = [3.5, 3.5, 3.5]
>>> net = InputScale(input_scale, input_center)
>>> output = net(inputs).asnumpy()
>>> assert np.all(output[:, 0] <= 0.5) and np.all(output[:, 0] >= -0.5)
>>> assert np.all(output[:, 1] <= 1.0) and np.all(output[:, 0] >= -1.0)
>>> assert np.all(output[:, 2] <= 2.0) and np.all(output[:, 0] >= -2.0)