def with_rank

stelliferous@0.0.2.0 · tensor_view.bend:231

import stelliferous@0.0.2.0/tensor_view.bend as TensorView
import Base

231# The axes with leading axes of extent 1 added up to rank; the same elements.
232def with_rank(+axes: List<&2,Axis>,rank: Nat) -> List<&2,Axis>:
233  List.append(&2,Axis,List.replicate(Axis,Nat.sub(rank,List.length(&2,Axis,axes)),Axis{1,0}),axes)

Uses (7)

Used by (1)