inplace operation: softmax the given specific dim
(x: &mut NdArray, dim: i32)
source not stored for this graph (policy: none)