如何在TensorboardX中显示学习的矩阵？

Question

我正在使用pytorch对数据进行分类，我想可视化神经网络的学习矩阵。由于矩阵很大，因此我更喜欢用颜色而不是数字来显示它们。但是，我找不到正确的方法来执行此操作。有人将整个矩阵转换为字符串序列，但是我认为这不是一个好的解决方案：How to visualize a tensor summary in tensorboard

你有什么主意吗？ 谢谢。