def inverse_coordinates

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

import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import Base

77def inverse_coordinates(dimensions: List<&2, U32>, +axes: List<&2, U32>,
78  +coordinates: List<&2, U32>, +axis: U32) -> Maybe<&2, List<&2, U32>>:
79  match dimensions:
80    case Nil{}:
81      Some{Nil{}}
82    case dimension <> tail:
83      reorder_head(value => reorder_tail(value, inverse_coordinates(tail, axes, coordinates, U32.inc(axis))),
84        coordinate_for(axes, coordinates, axis))

Uses (7)

Used by (24)