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()

预期结果：