~/jendhub

def left_dims

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

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

226# The dimensions of each operand's gradient before it is reduced to the
227# operand's shape: the output's batch, then the operand's own matrix
228# dimensions.
229def left_dims(+plan: Plan) -> List<&2, U32>:
230  Plan{output, +batch, +rows, +inner, columns, l1, l2, r1, r2, o1, o2, +lv, rv} = plan
231  List.append(&2, U32, batch, matrix_dims(U32.from_nat(rows), lv, U32.from_nat(inner), False{}))

Uses (7)

Used by (1)