einsum_frule

Function einsum_frule 

Source
pub fn einsum_frule<'a, T>(
    subscripts: &'a str,
    primals: &'a [&'a Tensor<T>],
    tangents: &'a [Option<&'a Tensor<T>>],
) -> Result<Tensor<T>>
where T: EinsumRuntimeValue,
Expand description

Stateless forward-mode rule (JVP) for einsum over dense primals.