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