law mapped_flat
bendtorch-core@0.1.0.0 · src/tensor/transform_LAWS.bend:39
import bendtorch-core@0.1.0.0/src/tensor/transform_LAWS.bend as TransformLAWS
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
39law mapped_flat:
40 for -input: Shape.Shape
41 for -axes: List<&2, U32>
42 for -plan: Layout.Plan<input, axes>
43 for -i: U32
44 for +mapping: Layout.Mapping<input, axes, plan, i>
45 {Layout.source_index(input, axes, plan, i, mapping) ==
46 Layout.flatten_words(Shape.dimensions(input), Layout.input_words(input, axes, plan, i, mapping), 0) : U32}