mindspore.ops.ones
- mindspore.ops.ones(shape, dtype=None)[源代码]
创建一个值全为1的tensor。
警告
参数 shape 在后续版本中将不再支持Tensor类型的输入。
- 参数:
shape (Union[tuple[int], list[int], int, Tensor]) - 指定的shape。
dtype (
mindspore.dtype
) - 指定的数据类型,默认None
。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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]]])