MCPcopy Create free account
hub / github.com/pytorch/tutorials / vmap_recipe.py

File vmap_recipe.py

unstable_source/vmap_recipe.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1"""
2torch.vmap
3==========
4This tutorial introduces torch.vmap, an autovectorizer for PyTorch operations.

Callers

nothing calls this directly

Calls 2

modelFunction · 0.70
fFunction · 0.70

Tested by

no test coverage detected