如何可视化张量板中的分布？

Question

我的模型的输出是log softmax，它产生分类分布的对数。我想形象地看到这种分布。如果我使用tensorboard的'add_histogram'模块，那么它将显示我不想要的分布的分布。我想按原样查看输出分布。

# y_hat is the output of the model which has dimensions (batch_size, seq_len, 256)
ax_size = y_hat.view(-1, 256).size()[0]

# from log distribution to categorical probability distribution
P_out = torch.exp(torch.sum(y_hat.view(-1, 256), dim=0)/ax_size)

writer.add_histogram('Posterior Distribution', P_out.cpu().detach().numpy(), step)

我希望看到P_out像在张量板上一样。但是看到的输出就是这种分布的分布。