先用torch的框架算出来了一批数据,想要看看他的直方图,结果直接放进去,plt作图慢道怀疑人生。
t=cos_dist_with_tag[keep_real_term_mask]
cos_dist=t[:,keep_real_term_mask]
cos_dist_f=cos_dist.flatten()# torch.tensor变量
plt.figure(figsize=(16,8))
plt.subplot(1,2,1)
plt.title('cos unwhite dist')
plt.hist(cos_dist_f,bins=100,label='unwhite')
plt.legend()
原因
因为plt库不能直接转换tensor的变量!反之会统计成torch.tensor(0.9)即便是0.9一个值。
要先把tensor向量转化成至少numpy数据,就可以秒出结果。
tensor变量已经有方法叫.numpy(),可以直转。
t=cos_dist_with_tag[keep_real_term_mask]
cos_dist=t[:,keep_real_term_mask]
cos_dist_f=cos_dist.flatten().numpy()# torch.tensor变量 转化numpy
plt.figure(figsize=(16,8))
plt.subplot(1,2,1)
plt.title('cos unwhite dist')
plt.hist(cos_dist_f,bins=100,label='unwhite')
plt.legend()



