/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.PrimitiveWithInfer.rst
15 строк
801 B
huan
modify the error links
18 апр 2024, 05:11
18 апр 2024, 05:11
e6f4d8a
Код
Авторство
О чём код?
mindspore.ops.PrimitiveWithInfer ================================ .. py:class:: mindspore.ops.PrimitiveWithInfer(name) PrimitiveWithInfer是Python中的原语基类，在python中定义了跟踪推理的函数。 可以重写四个方法来定义Primitive的推断逻辑：__infer__()、infer_shape()、infer_dtype()和infer_value()。如果在Primitive中定义了__infer__()，则__infer__()的优先级最高。 如果未定义__infer__()，则可以定义infer_shape()和infer_dtype()来描述shape和类型的推断逻辑。infer_value()用于常量传播。 关于如何自定义算子，请查看 `自定义算子 <https://www.mindspore.cn/tutorials/experts/zh-CN/master/operation/op_custom.html>`_ 。 参数： - **name** (str) - 当前Primitive的名称。