Function melior::dialect::ods::linalg::softmax

source ·
pub fn softmax<'c>(
    context: &'c Context,
    result: &[Type<'c>],
    input: Value<'c, '_>,
    output: Value<'c, '_>,
    dimension: IntegerAttribute<'c>,
    location: Location<'c>
) -> SoftmaxOperation<'c>
Expand description

Creates a softmax operation.