def viewed

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:117

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

117# The plan of a gather or an index_select of `length` entries at an axis in
118# range.
119def viewed(+kind: Lookup, +given: Slice.View, +length: Nat) -> Plan:
120  Slice.View{+outer, +extent, +inner} = given
121  Plan{kind, outer, extent, inner, length}