mindspore.ops.FillV2

class mindspore.ops.FillV2[源代码]

创建一个Tensor,其shape由 shape 指定,其值则由 value 进行填充。

输入:
  • shape (Union[Tuple[int], Tensor[int]]) - 1-D Tensor或Tuple,指定了输出Tensor的shape。 其数据类型必须是int32或int64。

  • value (Tensor) - value 是零维Tensor,其值用于填充输出 y

输出:
  • y (Tensor) - Tensor,其shape和值如上所述。

异常:
  • TypeError - 如果 shape 不是1-D Tensor或Tuple。

  • TypeError - 如果 shape 的数据类型不是int32或者int64。

  • ValueError - 如果 value 不是零维Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> fillV2 = ops.FillV2()
>>> output = fillV2(Tensor([2, 3], mindspore.int32), Tensor(1, mindspore.float32))
>>> print(output)
[[1. 1. 1.]
 [1. 1. 1.]]
>>> output = fillV2(Tensor([3, 3], mindspore.int64), Tensor(0, mindspore.int32))
>>> print(output)
[[0 0 0]
 [0 0 0]
 [0 0 0]]