如何用Nd4J计算softmax导数？

Question

我正在尝试使用Nd4J框架计算softmax导数。

我尝试使用SoftMaxDerivative类。

org.nd4j.linalg.api.ops.impl.transforms.strict.SoftMaxDerivative

如下：

derivative = Nd4j.getExecutioner().exec(new SoftMaxDerivative(x));

运算符应该计算出softax导数，但结果与我的数值梯度检查不匹配...

我的数据（即x）是一个二维矩阵，其中每一行代表一个样本。