mindspore.ops.Fill

class mindspore.ops.Fill

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

输入：

• type (mindspore.dtype) - 指定输出Tensor的数据类型。数据类型只支持 bool_ 和 number 。

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

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

输出：

Tensor。

异常：

TypeError - shape 不是元组。

支持平台：

Ascend GPU CPU

样例：

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