law mapped_output

bendtorch-core@0.1.0.0 · src/tensor/transform_LAWS.bend:48

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

48law mapped_output:
49  for -input: Shape.Shape
50  for -axes: List<&2, U32>
51  for -plan: Layout.Plan<input, axes>
52  for -i: U32
53  for mapping: Layout.Mapping<input, axes, plan, i>
54  {Layout.output_words(input, axes, plan, i, mapping) ==
55    Layout.reverse_words(Layout.unflatten_words(Layout.reverse_words(Shape.dimensions(Layout.output(input, axes, plan))), i)) : List<&2, U32>}