У меня есть некоторые данные в pytorch на графическом процессоре, скажем,
x = torch.cuda.FloatTensor(500, 3).uniform_()
Есть ли способ отобразить это как облако точек (ala https://matplotlib.org/gallery/mplot3d/scatter3d.html) без предварительной передачи данных вCPU?
Я смотрел на такие фреймворки, как matplotlib, VisPy, mayavi и т. д. Но все они, кажется, хотят получать данные через ЦП и не могут обрабатывать указатель GPU, скажем.
Кто-нибудь знает хороший метод? Желательно, чтобы заговор был неблокирующим, чтобы я мог во время тренировки видеть в реальном времени x. Любые идеи?