Как нарисовать точечную диаграмму в Tensorboard Pytorch?

Предполагая, что мне нужен общий точечный график, нарисованный в TensorBoard, который рисует 1-й batch[:, 0], batch[:, 1] каждой эпохи.

Как это можно сделать в TensorBoard?

Старый аналогичный вопрос (январь 2017 г.) имеет обходной путь, но я надеюсь, что теперь (декабрь 2020 г.) у нас есть технология для реального решения.

Недостаточно моей попытки:

if self._current_epoch == 0:
    self.logger.experiment.add_scalars("epoch", {"batch": batch[:, 1]}, batch[:, 0])

Дает мне замечательную ошибку

assert(scalar.squeeze().ndim == 0), 'скаляр должен быть 0D'


person Gulzar    schedule 25.12.2020    source источник