mindspore.ops.ones
- mindspore.ops.ones(shape, dtype=None)[源代码]
创建一个值全为1的tensor。
警告
2.9.0(不含)之后版本非兼容性变更:shape 将改为 size,且 dtype 将改为关键字参数。当前版本中, shape 的 Tensor 类型输入也会逐步移除。
- 参数:
shape (Union[tuple[int], list[int], int, Tensor]) - 指定的shape。
dtype (
mindspore.dtype) - 指定数据类型,默认None。
- 返回:
Tensor
- 支持平台:
AscendGPUCPU
样例:
>>> import mindspore >>> mindspore.ops.ones(4) Tensor(shape=[4], dtype=Float32, value= [ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00, 1.00000000e+00]) >>> mindspore.ops.ones((2, 3)) Tensor(shape=[2, 3], dtype=Float32, value= [[ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00], [ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00]]) >>> mindspore.ops.ones(mindspore.tensor([1, 2, 3])) Tensor(shape=[1, 2, 3], dtype=Float32, value= [[[ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00], [ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00]]])