56238f0
1
2
3
4
5
6
7
8
import torch import torch.nn as nn from torch.nn.functional import scaled_dot_product_attention as attention
