Как выполнять операции tensordot после перестановки

#pytorch #permute #tensordot #pytorch #перестановка #tensordot Вопрос: У меня есть 2 тензора, A и B: A = torch.randn([32,128,64,12],dtype=torch.float64) B = torch.randn([64,12,64,12],dtype=torch.float64) C = torch.tensordot(A,B,([2,3],[0,1])) D = C.permute(0,2,1,3) # shape:[32,64,128,12] тензор…