BatchedMatmulLAWS.batch_of: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/tensor/batched_matmul_LAWS.bend as BatchedMatmulLAWS
Lines 308-314 of bendtorch-core@0.1.0.0/src/tensor/batched_matmul_LAWS.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/tensor/batched_matmul_LAWS.bend&start_line=308&end_line=314

import Base

# Every extent but the last two: the batch.
def batch_of(dims: List<&2, Nat>) -> List<&2, Nat>:
  match dims:
    case +x <> +y <> +z <> tail:
      x <> batch_of(y <> z <> tail)
    case rest:
      Nil{}

Uses (2):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=List
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat