当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.ndarray.Custom用法及代码示例


用法:

mxnet.ndarray.Custom(*data, **kwargs)

参数：

  • data：(NDArray[]) - 自定义运算符的输入数据。
  • op_type：(string) - 自定义运算符的名称。这是传递给的名称mx.operator.register注册操作符。
  • out：(NDArray, optional) - 输出 NDArray 来保存结果。

返回：

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

应用以前端语言(如 Python)实现的自定义运算符。

自定义运算符应覆盖所需的方法，例如 forward 和 backward 。自定义运算符必须先注册后才能使用。请在此处查看教程：https://mxnet.incubator.apache.org/api/faq/new_op

示例：

将名为 my_custom_operator 的自定义运算符应用于 input 。

>>> output = mx.symbol.Custom(op_type='my_custom_operator', data=input)

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.Custom。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。