TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.scaled_grouped_mm

tensorplay.nn.functional.scaled_grouped_mm(*args, **kwargs)[source]

tensorplay.nn.functional.scaled_dot_product_attention

scaled_dot_product_attention(query, key, value, attn_mask=None, dropout_p=0.0, is_causal=False, scale=None, backend=None) -> Tensor

tensorplay.nn.functional.scaled_mm

Next Page

Ask DeepWiki