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