tensorplay.scaled_dot_product_attention

tensorplay.scaled_dot_product_attention(query, key, value, attn_mask=None, dropout_p=0.0, is_causal=False, scale=None, enable_gqa=False)[source]