Softmax雅可比

Question

我有一个二维NumPy数组，并且我沿轴-1对数组执行了 softmax操作。现在，在执行反向传播时，我必须计算softmax的导数。我对如何精确计算jacobian以及执行导数后获得的numpy数组的形状有些困惑。谁能帮我了解如何计算这种情况下的jacobian（Softmax操作已在2d numpy数组中沿轴-1执行）。