law axis_scatter_gather

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

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

77law axis_scatter_gather:
78  for +shape: Shape.Shape
79  for +axes: List<&2, U32>
80  for +coordinates: List<&2, U32>
81  for accepted: {True{} == Layout.bijective(shape, axes) : Bool}
82  for length: {Shape.rank_dimensions(coordinates) == Shape.rank_dimensions(Shape.dimensions(shape)) : Nat}
83  {Layout.inverse_coordinates(Shape.dimensions(shape), axes, Axis.order(U32, coordinates, Shape.naturals(axes)), 0) == Some{coordinates} : Maybe<&2, List<&2, U32>>}