forked from mindspore-Ecosystem/mindspore
38 lines
949 B
ReStructuredText
38 lines
949 B
ReStructuredText
|
mindspore.ops.Fill
|
|||
|
|
==================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.Fill(*args, **kwargs)
|
|||
|
|
|
|||
|
|
创建一个填充了Scalar值的Tensor。shape由 `shape` 参数指定，并用`value` 值填充该Tensor。
|
|||
|
|
|
|||
|
|
**输入：**
|
|||
|
|
|
|||
|
|
- **type** (mindspore.dtype) - 指定输出Tensor的数据类型。只支持常量值。
|
|||
|
|
- **shape** (tuple) - 指定输出Tensor的shape。只支持常量值。
|
|||
|
|
- **value** (scalar) - 用来填充输出Tensor的值。只支持常量值。
|
|||
|
|
|
|||
|
|
**输出：**
|
|||
|
|
|
|||
|
|
Tensor，shape为 `shape` 的值，值为 `value` 。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
**TypeError** - `shape` 不是元组。
|
|||
|
|
|
|||
|
|
**支持平台：**
|
|||
|
|
|
|||
|
|
``Ascend`` ``GPU`` ``CPU``
|
|||
|
|
|
|||
|
|
**样例：**
|
|||
|
|
|
|||
|
|
>>> fill = ops.Fill()
|
|||
|
|
>>> output = fill(mindspore.float32, (2, 2), 1)
|
|||
|
|
>>> print(output)
|
|||
|
|
[[1. 1.]
|
|||
|
|
[1. 1.]]
|
|||
|
|
>>> output = fill(mindspore.float32, (3, 3), 0)
|
|||
|
|
>>> print(output)
|
|||
|
|
[[0. 0. 0.]
|
|||
|
|
[0. 0. 0.]
|
|||
|
|
[0. 0. 0.]]
|
|||
|
|
|