GatherLAWS.entry_at: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/tensor/gather_LAWS.bend as GatherLAWS
Lines 228-235 of bendtorch-core@0.1.0.0/src/tensor/gather_LAWS.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_LAWS.bend&start_line=228&end_line=235

import Base
import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

# The entry output cell (o, k, r) of [outer, length, inner] reads: entry k
# of an index_select's index, the entry at the cell itself of a gather's.
def entry_at(+kind: Gather.Lookup, +count: Nat, +inner: Nat, +o: Nat, +k: Nat, +r: Nat) -> Nat:
  match kind:
    case Gather.Selected{}:
      k
    case Gather.Gathered{}:
      position(count, inner, o, k, r)

Uses (3):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=Lookup
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_LAWS.bend&name=position