python - 如何写光栅图？

Question

我在 google colab 上使用 pytorch。
我在下面有一个张量矩阵，这是示例，实际上矩阵大小约为 50 个神经元和 30,000~50,000 次。

a= torch.tensor([[0., 0., 0., 0., 0.],
                 [0., 0., 0., 0., 1.],
                 [0., 1., 0., 1., 0.]])

每个值是，

a= torch.tensor([[Neuron1(t=1), N2(t=1), N3(t=1), N4(t=1), N5(t=1)],
                 [N1(t=2), N2(t=2), N3(t=2), N4(t=2), N5(t=2)],
                 [N1(t=3), N2(t=3), N3(t=3), N4(t=3), N5(t=3)]])

1 表示神经元激活，0 表示不激活。
所以 Neuron5(t=2) , Neuron2(t=3)和 Neuron4(t=3)正在射击。
然后，我想使用此矩阵制作如下所示的栅格图或散点图，
点显示了放电神经元。

神经元数
1|
2| *
3|
4| *
5|__ *_____时间
1 2 3
什么是最好的python代码来做到这一点？
我现在不知道。
感谢您的阅读。

Answer 1

您可以按如下方式轻松完成:

import matplotlib.pyplot as plt
a= torch.tensor([[0., 0., 0., 0., 0.],
                 [0., 0., 0., 0., 1.],
                 [0., 1., 0., 1., 0.]],device='cuda')
plt.scatter(*torch.where(a.cpu()))