mindspore.Tensor.expand_as

查看源文件
mindspore.Tensor.expand_as(other)

将输入张量的shape扩展为另一个输入张量的shape。 输出张量的维度必须遵守广播规则,即输入张量的shape维度小于或者等于另一个输入张量的shape维度。

参数:
  • other (Tensor) - 目标张量。其shape为输入张量扩展的目标shape。

返回:

维度与另一个输入张量 other 的相同的Tensor,且数据类型与输入张量 self 相同。

异常:
  • TypeError - 如果另一个输入 other 不是张量。

  • ValueError - 如果 selfother 的shape不兼容。

支持平台:

Ascend

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1, 2, 3], [1, 2, 3]]).astype(np.float32))
>>> other = Tensor(np.array([[1, 1, 1], [1, 1, 1]]).astype(np.float32))
>>> output = x.expand_as(other)
>>> print(output)
[[1. 2. 3.]
 [1. 2. 3.]]
mindspore.Tensor.expand_as(x)

将输入张量的维度扩展为目标张量的维度。

参数:
  • x (Tensor) - 目标张量。其shape必须符合扩展的规则。

返回:

维度与目标张量的相同的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> input = Tensor([1, 2, 3], dtype=mstype.float32)
>>> x = Tensor(np.ones((2, 3)), dtype=mstype.float32)
>>> output = input.expand_as(x=x)
>>> print(output)
[[1. 2. 3.]
 [1. 2. 3.]]