Pytorch中使用tensorboard中如何添加低维映射add_embedding

这篇文章给大家介绍Pytorch中使用tensorboard中如何添加低维映射add_embedding，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

pytorch 中使用tensorboard，详解writer.add_embedding函数的作用

代码展示:

from torch.utils.tensorboard import SummaryWriterimport keywordimport torch
writer = SummaryWriter()meta = []while len(meta)<100:meta = meta+keyword.kwlist # get some stringsmeta = meta[:100]for i, v in enumerate(meta):meta[i] = v+str(i)label_img = torch.rand(100, 3, 10, 32)for i in range(100):label_img[i]*=i/100.0writer.add_embedding(torch.randn(100, 5), metadata=meta, label_img=label_img)# writer.add_embedding(torch.randn(100, 5), label_img=label_img)# writer.add_embedding(torch.randn(100, 5), metadata=meta)writer.close()# tensorboard --logdir=runs

运行结果(浏览器的页面需要刷新一下):

关于Pytorch中使用tensorboard中如何添加低维映射add_embedding就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。