mindspore.mint.full

查看源文件
mindspore.mint.full(size, fill_value, *, dtype=None)[源代码]

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

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

  • fill_value (Union(number.Number, Tensor)) - 用来填充输出Tensor的值。可以是一个标量值、0-D的Tensor或只有单个元素的1-D的Tensor。

关键字参数:
  • dtype (mindspore.dtype) - 指定输出Tensor的数据类型。数据类型只支持 bool_number ,更多细节详见 mindspore.dtype 。默认值: None

返回:

Tensor。

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

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

支持平台:

Ascend

样例:

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