~/jendhub

def transpose_reference

bendtorch-core@0.1.0.0 · src/tensor/transform_totality.bend:116

import bendtorch-core@0.1.0.0/src/tensor/transform_totality.bend as TransformTotality
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
import bendtorch-core@0.1.0.0/src/tensor/transpose_spec.bend as Transpose

116def transpose_reference(+input: Shape.Shape, +output: Shape.Shape, +a: U32, +b: U32, index: U32) -> Maybe<&2, U32>:
117  Layout.reference_source_coordinates(Shape.dimensions(input),
118    Axis.scatter(U32, Shape.rank_dimensions(Shape.dimensions(input)),
119      Shape.naturals(Transpose.requested_axes(Shape.rank_dimensions(Shape.dimensions(input)), U32.to_nat(a), U32.to_nat(b))),
120      Shape.words(Shape.digits(Shape.extents(output), U32.to_nat(index))), 0n))

Uses (13)

Used by (4)