mx.symbol.Custom

Description

Apply a custom operator implemented in a frontend language (like Python).

Custom operators should override required methods like forward and backward. The custom operator must be registered before it can be used. Please check the tutorial here: https://mxnet.incubator.apache.org/api/faq/new_op

Usage

mx.symbol.Custom(...)

Arguments

Argument

Description

data

NDArray-or-Symbol[].

Input data for the custom operator.

op.type

string.

Name of the custom operator. This is the name that is passed to mx.operator.register to register the operator.

name

string, optional.

Name of the resulting symbol.