如何绘制张量？

Question

我必须绘制维度为 [9,224,224] 的火炬张量。 有matplotlib.pyplot的方法吗？

Answer 1

您可以绘制图像

矩阵很容易绘制成图像。然后如果你有更高的维度，它们中的每一个都可以分解成一些矩阵。

a = np.random.random((9, 224, 224)).reshape((3, 3, 224, 224))
fig, ax = plt.subplots(3, 3)

for i in range(3):
    for j in range(3):
        ax[i][j].imshow(a[i][j])

值得考虑这是否适合所讨论的数据，但如果配置合适，它可以很好地工作。例如查看神经网络的权重。