PyTorch では CHW 形式になっているので、
単純に CHW -> HWC にすればプロットできます。

import matplotlib.pyplot as plt
# imgのshapeは(C,H W)
img = img.permute(1,2,0) # CHW -> HWC
plt.imshow(img)