Pytorch Tensor 的 Matplot 直方图

Question

我有一个大小为 10 的张量，只有 2 值：0 和 1。

我想简单地使用 matplotlib.pyplot.hist 绘制上面张量的直方图。这是我的代码：

import torch
import matplotlib.pyplot as plt
t = torch.tensor([0., 1., 1., 1., 0., 1., 1., 0., 0., 0.])
print(t)
plt.hist(t, bins=2)
plt.show()

输出：

为什么直方图中有这么多值？其余的值从哪里来？如何为我的张量绘制正确的直方图？

Answer 1

plt.hist(t, bins=2) 函数不适用于张量。为了使其正常工作，您可以尝试使用 t.numpy() 或 t.tolist() 代替。据我自学，使用 pytorch 计算直方图的方法是通过 torch.histc() 函数并使用 plt.bar() 绘制直方图函数如下：

import torch
import matplotlib.pyplot as plt

t = torch.tensor([0., 0., 1., 1., 0., 1., 1., 0., 0., 0.])
hist = torch.histc(t, bins = 2, min = 0, max = 1)

bins = 2
x = range(bins)
plt.bar(x, hist, align='center')
plt.xlabel('Bins')

可以看到一些绘制直方图的来源 此处 和 这里。我找不到根本原因，如果有人可以教育我，那就太好了，但据我所知，这是绘制张量的方法，

我将张量更改为 4 '1.0' 和 6 '0.0 ' 能够看到差异