pytorch 1.0将tensor 转化为 to numpy
a是tensor
a.data.cpu().numpy()
plt.switch_backend('agg')
after import matplotlib.pyplot as plt
raise RuntimeError('Invalid DISPLAY variable')
pytorch 1.0将tensor 转化为 to numpy
a是tensor
a.data.cpu().numpy()
plt.switch_backend('agg')
after import matplotlib.pyplot as plt
raise RuntimeError('Invalid DISPLAY variable')