torch.vmap ========== .. currentmodule:: torch .. autofunction:: vmap