def broadcast_index

bendtorch-core@0.1.0.0 · src/shape/core.bend:182

import Base

182def broadcast_index(size: Nat, index: Nat) -> Nat:
183  match size:
184    case 0n:
185      index
186    case 1n+p:
187      match p:
188        case 0n:
189          0n
190        case 1n+q:
191          index