def taken

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
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

83def taken(+kind: Gather.Lookup, +dimensions: List<&2, U32>, +at: Nat, +count: Nat) -> Result<&2, &2, Gather.Error, Gather.Planned>:
84  output(Slice.resized(dimensions, at, U32.from_nat(count)),
85    viewed(kind, Slice.sized(dimensions, at, Slice.holds_zero(dimensions)), count),
86    Shape.construct(Slice.resized(dimensions, at, U32.from_nat(count))))