def bmap

bendtorch@1.3.0.0 · tensor.bend:144

import Base

144# reading a (any rank up to o's) along o
145def bmap(+a: List<&2, Nat>, +o: List<&2, Nat>) -> IMap:
146  Strided{0, List.reverse(&2, Ax, axes(List.reverse(&2, Nat, pad(a, List.length(&2, Nat, o))), List.reverse(&2, Nat, o), 1, Nil{}))}