law axis_scatter_complete

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

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

68law axis_scatter_complete:
69  for +shape: Shape.Shape
70  for +axes: List<&2, U32>
71  for +coordinates: List<&2, U32>
72  for accepted: {True{} == Layout.bijective(shape, axes) : Bool}
73  for length: {Shape.rank_dimensions(coordinates) == Shape.rank_dimensions(Shape.dimensions(shape)) : Nat}
74  {Layout.inverse_coordinates(Shape.dimensions(shape), axes, coordinates, 0) ==
75    Some{Axis.order(U32, coordinates, Axis.inverse_axes(Shape.rank_dimensions(Shape.dimensions(shape)), Shape.naturals(axes)))} : Maybe<&2, List<&2, U32>>}