Table Of Contents
Table Of Contents

mx.nd.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: http://mxnet.io/faq/new_op.html.

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.