def mapping_output_decoded

bendtorch-core@0.1.0.0 · src/tensor/transform_PROOF.bend:2056

import bendtorch-core@0.1.0.0/src/tensor/transform_PROOF.bend as TransformPROOF
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

2056def mapping_output_decoded(+input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +index: U32,
2057  mapping: Layout.Mapping<input, axes, plan, index>) ->
2058  {Layout.output_words(input, axes, plan, index, mapping) == Layout.reverse_words(Layout.unflatten_words(Layout.reverse_words(Shape.dimensions(Layout.output(input, axes, plan))), index)) : List<&2, U32>}:
2059  Layout.Mapped{oi, oc, ic, source, requested, decoded, inverse, flattened, roundtrip} = mapping
2060  decoded

Uses (10)

Used by (2)