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.