mindspore.ops.full

mindspore.ops.full(size, fill_value, *, dtype=None)[源代码]

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

参数：

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

  • fill_value (number.Number) - 用来填充输出Tensor的值。当前不支持复数类型。

关键字参数：

  • dtype (mindspore.dtype) - 指定输出Tensor的数据类型。数据类型只支持 bool_ 和 number ，更多细节详见 mindspore.dtype 。默认值： None 。

返回：

Tensor。

异常：

  • TypeError - size 不是元组或列表。

  • ValueError - size 中包含小于0的成员。

支持平台：

Ascend GPU CPU

样例：

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