def viewed

bendtorch-core@0.1.0.0 · src/tensor/gather_LAWS.bend:67

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

67# The plan of a request the checks take: the row-major view of the input
68# around the axis, slice.bend's, and the length of the output's axis ...
69def viewed(+kind: Gather.Lookup, +given: Slice.View, +count: Nat) -> Gather.Plan:
70  Slice.View{+outer, +rows, +inner} = given
71  Gather.Plan{kind, outer, rows, inner, count}