def left_layout

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:212

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix

212# g @ right^T, the left operand's gradient over the output's batch: g
213# [.., rows, columns] by rows, right [.., inner, columns] by its rows, the
214# columns being the terms.
215def left_layout(+plan: Plan) -> Layout:
216  Plan{output, batch, +rows, +inner, +columns, l1, l2, +r1, +r2, +o1, +o2, lv, rv} = plan
217  Layout{Matrix.Kept{}, Matrix.Kept{}, columns, rows, inner, o1, o2, r1, r2, o2}