mindspore.nn.Identity

class mindspore.nn.Identity[源代码]

返回与输入具有相同shape和值的Tensor。

输入:
  • x (Tensor) - shape为 \((x_1, x_2, ..., x_R)\) 的Tensor。数据类型为Number。

输出:

Tensor,shape和数据类型与 x 相同。

异常:
  • TypeError - x 不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int64)
>>> net = nn.Identity()
>>> output = net(x)
>>> print(output)
[1 2 3 4]