def right_slot

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:329

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

329def right_slot(-left: List<&2, Nat>, -right: List<&2, Nat>, -output: List<&2, Nat>, map: Shape.Index<left, right, output>) -> Slot:
330  match map:
331    case Shape.Indexed{+rows, exact}:
332      slot_of(same_right(rows), zero_right(rows))
333    case Shape.Unindexed{zero}:
334      WalkedSlot{}