Python PyTorch SummaryWriter.add_histogram用法及代碼示例

本文簡要介紹python語言中 torch.utils.tensorboard.writer.SummaryWriter.add_histogram 的用法。

用法:

add_histogram(tag, values, global_step=None, bins='tensorflow', walltime=None, max_bins=None)

參數：

• tag(string) -數據標識符

• values(torch.Tensor,numpy.array， 或者字符串/blob 名稱) -建立直方圖的值

• global_step(int) -要記錄的全局步長 值

• bins(string) -{‘tensorflow’,'auto', ‘fd’, ...} 之一。這決定了如何製作箱子。您可以在以下位置找到其他選項：https://docs.scipy.org/doc/numpy/reference/generated/numpy.histogram.html

• walltime(float) -事件紀元後的可選覆蓋默認 walltime (time.time()) 秒

將直方圖添加到摘要中。

例子：

from torch.utils.tensorboard import SummaryWriter
import numpy as np
writer = SummaryWriter()
for i in range(10):
    x = np.random.random(1000)
    writer.add_histogram('distribution centers', x + i, i)
writer.close()

預期結果：