如何在Torch Tensorboard嵌入中添加多个元数据？

Question

我正在使用torch==1.9.0和tensorboard==2.5.0。我想用tensorbaord作为嵌入来跟踪数据，所以我是这样的：

data = np.random.poisson(lam=10.0, size=(4,4))
labels = ["A","A","B","B"]
ids = [1,2,3,4]
writer = SummaryWriter("/runs/")
writer.add_embedding(data,
                    metadata=labels)
writer.close()

但我只能将标签或ID添加为metadeta，而不能添加组合的词汇。有没有办法，怎么解决这个问题？谢谢

仅供参考：Tensorboard文档只是将CIRBES元数据描述为一个列表： https://pytorch.org/docs/stable/tensorboard.html

Answer 1

找到了答案。您可以通过添加元数据头获得多个字段，并将metadat作为列表列表提供：

writer.add_embedding(data.values,
                    metadata=metadata,
                    metadata_header=["Name","Labels"])

参考：https://github.com/tensorflow/tensorboard/issues/61