mindspore.ops.fill
- mindspore.ops.fill(type, shape, value)[源代码]
创建一个指定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
样例:
>>> output = ops.fill(mindspore.float32, (2, 2), 1) >>> print(output) [[1. 1.] [1. 1.]] >>> output = ops.fill(mindspore.float32, (3, 3), 0) >>> print(output) [[0. 0. 0.] [0. 0. 0.] [0. 0. 0.]]