def batch_view

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

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

378def batch_view(+plan: Batched.Plan) -> List<&2, Nat>:
379  Batched.Plan{output, batch, rows, inner, columns, l1, l2, r1, r2, +o1, +o2, lv, rv} = plan
380  [o1, o2]

Uses (3)

Used by (1)