def right_operand

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

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

62def right_operand(dims: List<&2, U32>) -> Result<&2, &2, Matrix.Error, Operand>:
63  match dims:
64    case +k <> Nil{}:
65      Done{Operand{Batch{0n, 1, 1}, 1, k, True{}}}
66    case +k <> +n <> Nil{}:
67      Done{Operand{Batch{0n, 1, 1}, n, k, False{}}}
68    case +b <> +k <> +n <> Nil{}:
69      Done{Operand{Batch{1n, 1, b}, n, k, False{}}}
70    case +a <> +b <> +k <> +n <> Nil{}:
71      Done{Operand{Batch{2n, a, b}, n, k, False{}}}
72    case rest:
73      Fail{Matrix.RankMismatch{}}