mindspore.ops.fill

mindspore.ops.fill(type, shape, value)[源代码]

创建一个指定shape的Tensor,并用指定值填充。

参数:
  • type (mindspore.dtype) - 指定输出Tensor的数据类型。数据类型只支持 bool_number

  • shape (Union(Tensor, tuple[int])) - 指定输出Tensor的shape。

  • value (Union(Tensor, number.Number, bool)) - 用来填充输出Tensor的值。

返回:

Tensor。

异常:
  • TypeError - shape 不是元组或Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import ops
>>> output = ops.fill(mindspore.float32, (2, 2), 1)
>>> print(output)
[[1. 1.]
 [1. 1.]]
>>> output = ops.fill(mindspore.float32, (3, 3), 0)
>>> print(output)
[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]