python - 如何绘制张量？

Question

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

标签： pythonmatplotlibtorchtensor

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])

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