问题描述

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)

输出正常，无省略号，解决撒花。