Вывод моей модели - это журнал softmax, который выдает журнал категориального распределения. Я хотел бы представить это распределение как есть. Если бы я использовал модуль addor_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 как на тензорной доске. Но видимый результат - это распределение этого распределения.