Try this and show here the results:
print(epoch_accuracy_o)
print(epoch_accuracy_o.shape)




 thida:

[h.cpu().numpy() for h in epoch_accuracy_o]


here h is a numpy array, but it must be pytorch tensor…