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