Function einsum_codegen::codegen::ndarray::naive::contraction
source · pub fn contraction(subscripts: &Subscripts) -> TokenStream
Expand description
Generate naive contraction loop
for i in 0..n_i {
for k in 0..n_k {
for j in 0..n_j {
out0[(i, k)] = arg0[(i, j)] * arg1[(j, k)];
}
}
}