law axis_gather_refines

bendtorch-core@0.1.0.0 · src/tensor/axis_LAWS.bend:19

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis

19law axis_gather_refines:
20  for +axes: List<&2, U32>
21  for +items: List<&2, U32>
22  {Layout.reorder(items, axes) == Axis.gather(U32, items, Shape.naturals(axes)) : Maybe<&2, List<&2, U32>>}