问题描述

python3 load了pkl文件后，发现打印出来的数组有省略号，不能用于继续的计算和操作:

import pickle
with 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 np

np.set_printoptions(threshold=np.inf)

如果是tensor的话，在脚本开始加这两行即可

import torch

torch.set_printoptions(threshold=np.inf)

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