二维图上点的和分布

Question

我在pytorch张量中有10000个二维点。这些点介于-1和1之间。即x ∈ {-1, 1}^2我想输出一个热图来显示点的分布。为了打印图形，我正在使用类似的matplotlib

torch.meshgrid(torch.linspace(-1, 1, 1000), torch.linspace(-1, 1, 1000))
ax.contourf(x_grid, y_grid, grid_values)

然而，我不知道如何计算网格值在给定图上点的形状张量(10000，2(的情况下，我想计算网格值(注意，一个点可以在图上出现多次，我们想求出这些点的分布的和(

Answer 1

您正在描述一个正常的2D直方图，可能是加权的(不确定您的点是否也有相关值(？

plt.hist2d(x, y, weights=w, range=[[-1,1],[-1,1]], bins=1000)

还将吐出相关的矩阵，并相应地对值进行装箱，您可以使用这些值来制作轮廓。