/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.func_kernel.rst
17 строк
1 KB
shaojunsong
fix docs r2.3q1
18 мар 2024, 15:29
18 мар 2024, 15:29
45407cf
Код
Авторство
О чём код?
mindspore.ops.kernel ======================= .. py:function:: mindspore.ops.kernel(fn=None, reg_info=None, compile_attrs=None) 用于MindSpore Hybrid DSL函数书写的装饰器。 给用MindSpore Hybrid DSL书写的函数加上此装饰器后，它可以用作一个普通的Python函数。 与此同时，他可以用于自定义算子 :class:`mindspore.ops.Custom` （参数 `func_type` 可以设置为 ``"hybrid"`` 或者 ``"pyfunc"``）的输入 。 使用 `hybrid` 类型的 :class:`mindspore.ops.Custom` 自定义算子可以自动推导数据类型和形状。 参数： - **fn** (Function) - 将被作为自定义算子运行的Python函数。默认值： ``None`` 。 - **reg_info** (tuple[str, dict]) - 包含算子注册信息的dict或json字符串。默认值： ``None`` 。 - **compile_attrs** (Dict) - 算子编译信息。默认值： ``None`` 。 返回： Function，如果 `fn` 不是None，那么返回一个用Hybrid DSL写的可执行函数；如果 `fn` 是None，则返回一个装饰器，该装饰器只有 `fn` 一个参数。