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.]]