def entry_same

bendtorch-core@0.1.0.0 · src/tensor/gather_PROOF.bend:355

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

355# The arithmetic of the positions: output cell (o, k, r) reads the entry
356# the laws name and the input cell (o, j, r).
357def entry_same(+kind: Gather.Lookup, +count: Nat, +inner: Nat, +o: Nat, +k: Nat, +r: Nat) ->
358  {Gather.entry_of(kind, count, inner, o, k, r) == Laws.entry_at(kind, count, inner, o, k, r) : Nat}:
359  match kind:
360    case Gather.Selected{}:
361      {==}
362    case Gather.Gathered{}:
363      {==}