did you np.allclose(output_teacher_batch.data.cpu(), output_teacher_batch.data.cpu().numpy()) ? if yes, I tried it and it return true. so I guess it means these two have equal values, but how come I comparing them myself they are not the same