如何在DJL (Deep Java Library)中调用自定义mxnet运算符？

import ai.djl.ndarray.NDArray;
import ai.djl.ndarray.NDManager;
import ai.djl.nn.AbstractBlock;
import ai.djl.nn.Block;
import ai.djl.nn.Parameter;
import ai.djl.training.ParameterStore;
import ai.djl.util.Utils;

public class CustomOperatorBlock extends AbstractBlock {

    private Parameter weight;

    public CustomOperatorBlock() {
        weight = addParameter(new Parameter("weight"));
    }

    @Override
    public NDArray forward(ParameterStore parameterStore, NDList inputs, boolean training) {
        NDManager manager = inputs.get(0).getManager();
        NDArray input = inputs.singletonOrThrow();
        
        // 调用自定义运算符
        NDArray output = customOperator(input, weight.getValue().toTensor());

        return output;
    }

    private NDArray customOperator(NDArray input, NDArray weight) {
        // 自定义运算符的计算逻辑
        NDArray output = input.mul(weight);

        return output;
    }
}