Table Of Contents
Table Of Contents

Custom

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

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.

Defined in src/operator/custom/custom.cc:L547

Parameters:
  • data (NDArray[]) – 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.
  • out (NDArray, optional) – The output NDArray to hold the result.
Returns:

out – The output of this function.

Return type:

NDArray or list of NDArrays

Example

Applies a custom operator named my_custom_operator to input.

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