问题描述
python3 load了pkl文件后,发现打印出来的数组有省略号,不能用于继续的计算和操作:
import picklewith open('./filename.pkl','rb') as f:data=pickle.load(f)print(data)
输出长这样:
[[0. 1. 0. ... 0. 0. 0.][0. 0. 0. ... 0. 0. 0.][0. 1. 0. ... 0. 0. 0.]...[0. 0. 0. ... 1. 0. 0.][0. 0. 0. ... 1. 0. 0.][0. 0. 0. ... 1. 0. 0.]]
原因分析:
数组长度太长
解决方案:
在脚本开始加这两行即可
import numpy as npnp.set_printoptions(threshold=np.inf)
如果是tensor的话,在脚本开始加这两行即可
import torchtorch.set_printoptions(threshold=np.inf)
输出正常,无省略号,解决撒花。