pytorch中tensor变量变为numpy的数组变量并保存为excel.csv文件,进行可视化

import torch
import numpy as np

a = torch.ones(5)
print(a)

b = a.numpy()
print(b)

np.savetxt("test.csv",b)
# np.savetxt("score.csv",scores,delimiter=",",header="英语,数学",comments="",fmt="%d")
# 英语 数学之间的符号应该和delimiter符号一致
# help(np.savetxt)