保存tensor到csv中

1 首先将GPU上的tensor复制一封到cpu上：

save_occ=occ_mask.data.cpu()

2 再将高纬度tensor转换到2维度的tensor,

save_occ=occ_mask.data.cpu()
with open("1.csv","ab") as f:
    # for i in range(occ_mask.shape[1]):
    # one_layer=occ_mask[1,:,:]
    np.savetxt(f,save_occ[0,:,:],delimiter=',',fmt="%d")

numpy.savetxt 参数详细参数见：

numpy.savetxt 参数_*Snowgrass*的博客-CSDN博客_numpy.savetxt