Как рассчитать производную высокого порядка от квадрата матрицы смежности?

Question

Я хотел бы рассчитать производные высокого порядка в Pytorch, но результаты - это нан.

X = torch.tensor([[[0.,0.,0.],[1.,0.,0.],[3.,0.,0.]]],requires_grad=True)
adjacency_matrix = (X.unsqueeze(2)-X.unsqueeze(1)).norm(dim=3)
Y = torch.matmul(adjacency_matrix,adjacency_matrix)
grad1 = torch.autograd.grad(Y.sum(),X,create_graph=True)[0]
grad2 = torch.autograd.grad(grad1.sum(),X,create_graph=True)[0]
print(grad2)

Выходной сигнал grad2 - это все нан.Подскажите, пожалуйста, как получить правильный результат?