mindspore.ops.full_like
- mindspore.ops.full_like(input, fill_value, *, dtype=None)[源代码]
返回一个shape与 input 相同并且使用 fill_value 填充的Tensor。
- 参数:
input (Tensor) - 输入Tensor,输出Tensor与 input 具有相同的shape。
fill_value (Number) - 用来填充输出Tensor的值。当前不支持复数类型。
- 关键字参数:
dtype (mindspore.dtype, 可选) - 指定输出Tensor的数据类型。数据类型只支持 bool_ 和 number ,更多细节详见
mindspore.dtype
。默认值:None
。
- 返回:
Tensor。
- 异常:
TypeError - input 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import Tensor, ops >>> input = Tensor([[0, 1], [2, 1]], dtype=mindspore.int32) >>> output = ops.full_like(input, 1) >>> print(output) [[1. 1.] [1. 1.]] >>> input = Tensor([[0, 1, 1], [2, 1, 2], [1, 3, 4]], dtype=mindspore.int32) >>> output = ops.full_like(input, 0) >>> print(output) [[0. 0. 0.] [0. 0. 0.] [0. 0. 0.]]