batch[:, 0], batch[:, 1]すべてのエポックの 1 番目を描画する TensorBoard で描画される一般的な散布図が必要であると仮定します。
TensorBoard でどのように行うことができますか?
古い同様の質問(2017 年 1 月) には回避策がありますが、現在 (2020 年 12 月) に実際のソリューションのための技術があることを願っています。
私の試みでは十分ではありません：
if self._current_epoch == 0:
self.logger.experiment.add_scalars("epoch", {"batch": batch[:, 1]}, batch[:, 0])
素晴らしいエラーが表示されます
assert(scalar.squeeze().ndim == 0), 'scalar should be 0D'