Skip to content

Support applying Softmax over multidimensional tensors

Javier Duarte requested to merge github/fork/vloncar/multidim_softmax into master

Created by: vloncar

Current implementations of softmax are incorrect when applied to multidimensional tensors (used in e.g., ENet model). The implementations in this PR handle this properly for axis=-1 (the default).

Merge request reports