It seems that I get errors unless I put my tensors on cpu instead of my cuda device? Should I be putting my tensors on cpu for matplotlib?
matplotib expects numpy arrays as its input, which are only supported on the CPU.
It seems that I get errors unless I put my tensors on cpu instead of my cuda device? Should I be putting my tensors on cpu for matplotlib?
matplotib expects numpy arrays as its input, which are only supported on the CPU.