这篇文章给大家介绍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就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
