當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。