def left_view

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_LAWS.bend:367

import Base
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as Batched

367# What a plan holds, read back as lists: the left operand as
368# [l1, l2, rows, inner], the right one as [r1, r2, inner, columns], the
369# output's batch as [o1, o2], and which operands are vectors.
370def left_view(+plan: Batched.Plan) -> List<&2, Nat>:
371  Batched.Plan{output, batch, +rows, +inner, columns, +l1, +l2, r1, r2, o1, o2, lv, rv} = plan
372  [l1, l2, rows, inner]