使用pytorch，如何在损失函数中写一个雅可比矩阵？

Question

在知识蒸馏中，我们通常使用像这样的雅可比匹配：T and S are the logits produced by the teacher network and student network

非常感谢！