tensorplay.special._elementwise.log_softmax

tensorplay.special._elementwise.log_softmax(input, dim=None, *, dtype=None)[source]

log(softmax(input, dim)), evaluated without the intermediate exp.