pub trait SoftmaxAxis: SoftmaxActivation {
// Required method
fn softmax_axis(self, axis: usize) -> Self::Output;
}Expand description
Compute the softmax activation along a specified axis.
pub trait SoftmaxAxis: SoftmaxActivation {
// Required method
fn softmax_axis(self, axis: usize) -> Self::Output;
}Compute the softmax activation along a specified axis.