Skip to content

add vmap for mapping a function over a leading dimension #2986

add vmap for mapping a function over a leading dimension

add vmap for mapping a function over a leading dimension #2986