|
|
|
@@ -12,42 +12,3 @@ mindspore.ops.PrimitiveWithInfer |
|
|
|
**参数:** |
|
|
|
|
|
|
|
- **name** (str) - 当前Primitive的名称。 |
|
|
|
|
|
|
|
.. py:method:: infer_dtype(*args) |
|
|
|
|
|
|
|
根据输入类型推断输出类型。 |
|
|
|
|
|
|
|
**参数:** |
|
|
|
|
|
|
|
- **args** (:class:`mindspore.dtype`) - 输入的数据类型。 |
|
|
|
|
|
|
|
**返回:** |
|
|
|
|
|
|
|
:class:`mindspore.dtype`,输出的数据类型。 |
|
|
|
|
|
|
|
.. py:method:: infer_shape(*args) |
|
|
|
|
|
|
|
根据输入形状推断输出形状。 |
|
|
|
|
|
|
|
.. note:: |
|
|
|
Scalar的shape是一个空元组。 |
|
|
|
|
|
|
|
**参数:** |
|
|
|
|
|
|
|
- **args** (tuple(int)) - 输入tensor的shape。 |
|
|
|
|
|
|
|
**返回:** |
|
|
|
|
|
|
|
`tuple(int)`,输出tensor的shape。 |
|
|
|
|
|
|
|
.. py:method:: infer_value(*args) |
|
|
|
|
|
|
|
根据编译时的输入值推断输出值。 |
|
|
|
|
|
|
|
**参数:** |
|
|
|
|
|
|
|
- **args** (Any) - 输入的值。 |
|
|
|
|
|
|
|
**返回:** |
|
|
|
|
|
|
|
输出的值。如果编译时无法推断该值,返回 `None` 。 |