我在 GPU 上的 pytorch 中有一些数据，比如说，
x = torch.cuda.FloatTensor(500, 3).uniform_()
有没有办法在不先将数据传输到 CPU 的情况下将其渲染为点云（ala https://matplotlib.org/gallery/mplot3d/scatter3d.html ）？
我看过 matplotlib、VisPy、mayavi 等框架。但它们似乎都希望通过 CPU 获取数据，并且无法处理 GPU 指针。
有谁知道好的方法吗？最好，绘图应该是非阻塞的，这样我可以x在训练期间实时查看。有任何想法吗？