mindelec.architecture.InputScaleNet

class mindelec.architecture.InputScaleNet(input_scale, input_center=None)

将输入值缩放到指定的区域。

参数：

• 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

样例：

>>> import numpy as np
>>> from mindelec.architecture import InputScaleNet
>>> 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 = InputScaleNet(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)