law axis_scatter_refines

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

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

24law axis_scatter_refines:
25  for +shape: Shape.Shape
26  for +axes: List<&2, U32>
27  for +coordinates: List<&2, U32>
28  {Layout.inverse_coordinates(Shape.dimensions(shape), axes, coordinates, 0) ==
29    Axis.scatter(U32, Shape.rank_dimensions(Shape.dimensions(shape)), Shape.naturals(axes), coordinates, 0n) : Maybe<&2, List<&2, U32>>}