溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章給大家介紹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進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
