mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Updated Autograd Onboarding Lab (markdown)
@ -8,7 +8,7 @@ You should add both your mentor as well as @albanD as a reviewer for this PR.
|
||||
|
||||
For this lab, we are going to consider the following attention function
|
||||
|
||||
```
|
||||
```py
|
||||
import torch
|
||||
|
||||
def attn(q, k, v):
|
||||
|
Reference in New Issue
Block a user