It sadly, does not and the error is still presumably on this line:  G = torch.mm(feature_map, feature_map.t())
I did try to use  matmul()  instead but with the same result. I also tried to use other functions, but still with the same variables, and then it suddenly worked. So it really seems like the error is in  mm() …