pub fn softmax<'graph, A, F: Float>(x: A, axis: isize) -> Tensor<'graph, F>
where
    A: AsRef<Tensor<'graph, F>> + Copy,
{
    let x = x.as_ref();
    let g = x.graph();
    let op = activation_ops::Softmax { axis };
    Tensor::builder(g).append_input(x.as_ref(), false).build(op)
}
/// Computes softmax along specified axis
///
/// `axis` can be negative.
Computes softmax along specified axis
`axis` can be negative.
Computes softmax along specified axis ` axis ` can be negative .
pub fn softmax<'graph, A, F: Float>(x: A, axis: isize) -> Tensor<'graph, F>
where
    A: AsRef<Tensor<'graph, F>> + Copy,
{
    let x = x.as_ref();
    let g = x.graph();
    let op = activation_ops::Softmax { axis };
    Tensor::builder(g).append_input(x.as_ref(), false).build(op)
}
pub fn softmax < ' graph , A , F : Float > ( x : A , axis : isize ) -> Tensor < ' graph , F > where A : AsRef < Tensor < ' graph , F > > + Copy , { let x = x . as_ref ( ) ; let g = x . graph ( ) ; let op = activation_ops :: Softmax { axis } ; Tensor :: builder ( g ) . append_input ( x . as_ref ( ) , false ) . build ( op ) }
Computes softmax along specified axis
`axis` can be negative.
Computes softmax along specified axis ` axis ` can be negative .
x
A
axis
isize
x
A
axis
isize