Searched defs:log_softmax (Results 1 – 2 of 2) sorted by relevance
36 log_softmax = core.define_unary_op('log_softmax', nn.log_softmax) variable
2932 def log_softmax(logits, axis=None, name=None, dim=None): function