Function einsum_device

pub fn einsum_device<'a>(
    subscripts: &str,
    operands: impl IntoIterator<Item = &'a Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>

Perform the Einstein summation convention on the operands.

Params

• subscripts: Einstein summation convention equation
• operands: input arrays
• stream: stream or device to evaluate on