Skip to main content

softmax_axes

Macro softmax_axes 

Source
macro_rules! softmax_axes {
    ($a:expr, $axes:expr, $precise:expr) => { ... };
    ($a:expr, $axes:expr, $precise:expr, stream = $stream:expr) => { ... };
}
👎Deprecated since 0.26.0:

use with_stream or with_device around softmax_axes

Expand description

Macro generated for the function crate::ops::softmax_axes. See the function documentation for more details.