Skip to main content

scaled_dot_product_attention_device

Function scaled_dot_product_attention_device 

Source
pub fn scaled_dot_product_attention_device<'a>(
    queries: impl AsRef<Array>,
    keys: impl AsRef<Array>,
    values: impl AsRef<Array>,
    scale: f32,
    mask: impl IntoOption<ScaledDotProductAttentionMask<'a>>,
    sinks: impl Into<Option<&'a Array>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
👎Deprecated since 0.26.0:

use with_stream or with_device around scaled_dot_product_attention

Expand description

Compatibility shim for scaled_dot_product_attention.