def recipe

bendtorch-core@0.1.0.0 · src/tensor/permute_build.bend:77

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis

77def recipe(+request: Request) -> Recipe:
78  Request{+input, +axes} = request
79  +dims = Shape.dimensions(input)
80  Recipe{picks(Shape.rank_dimensions(dims), Shape.naturals(axes), Shape.naturals(Axis.order(U32, dims, Shape.naturals(axes))), 0n),
81    strides(Shape.naturals(dims))}