https://pytorch.org/docs/master/generated/torch.nn.functional.scaled_dot_product_attention.html