mindspore/docs/api/api_python/ops/mindspore.ops.func_ones.rst

18 lines
589 B
ReStructuredText
Raw Normal View History

mindspore.ops.ones
===================
.. py:function:: mindspore.ops.ones(shape, type)
创建一个值全为1的Tensor。
第一个参数指定Tensor的shape第二个参数指定填充值的数据类型。
2022-07-25 17:18:38 +08:00
参数:
- **shape** (Union[tuple[int], int]) - 指定输出Tensor的shape只能是正整数常量。
- **type** (mindspore.dtype) - 指定输出Tensor的数据类型只能是常量值。
2022-07-25 17:18:38 +08:00
返回:
Tensorshape和数据类型与输入相同。
2022-07-25 17:18:38 +08:00
异常:
- **TypeError** - `shape` 既不是tuple也不是int。