~/jendhub

def right_dims

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

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

233def right_dims(+plan: Plan) -> List<&2, U32>:
234  Plan{output, +batch, rows, +inner, +columns, l1, l2, r1, r2, o1, o2, lv, +rv} = plan
235  List.append(&2, U32, batch, matrix_dims(U32.from_nat(inner), False{}, U32.from_nat(columns), rv))

Uses (7)

Used by (1)