mindspore.ops.OnesLike
- class mindspore.ops.OnesLike[源代码]
返回值为1的Tensor,shape和数据类型与输入相同。
更多参考详见
mindspore.ops.ones_like()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> oneslike = ops.OnesLike() >>> input_x = Tensor(np.array([[0, 1], [2, 1]]).astype(np.int32)) >>> output = oneslike(input_x) >>> print(output) [[1 1] [1 1]]