frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated vmap Onboarding Lab (markdown)

Manuel
2024-07-03 19:35:28 +02:00
parent 2185127065
commit d63bdc403e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vmap-Onboarding-Lab.md

@ -89,4 +89,6 @@ std::tuple<Tensor,optional<int64_t>> simple_mul_batch_rule(
```
Avoid using the BINARY_POINTWISE macro (that would solve the problem trivially).
## Next
Unit 8: function transforms/Training Loops (Optional) - [[composable function transforms (aka torch.func, functorch)| function-transforms-(aka-torch.func,-functorch)]]