Skip to main content

scaled_dot_product_attention

Macro scaled_dot_product_attention 

Source
macro_rules! scaled_dot_product_attention {
    ($queries:expr, $keys:expr, $values:expr, $scale:expr) => { ... };
    ($queries:expr, $keys:expr, $values:expr, $scale:expr, stream = $stream:expr) => { ... };
    ($queries:expr, $keys:expr, $values:expr, $scale:expr, mask = $mask:expr) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, stream = $stream:expr
    ) => { ... };
    ($queries:expr, $keys:expr, $values:expr, $scale:expr, sinks = $sinks:expr) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, sinks =
        $sinks:expr, stream = $stream:expr
    ) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, sinks = $sinks:expr
    ) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, sinks = $sinks:expr, stream = $stream:expr
    ) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, sinks = $sinks:expr
    ) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, sinks = $sinks:expr, stream = $stream:expr
    ) => { ... };
}
👎Deprecated since 0.26.0:

use with_stream or with_device around scaled_dot_product_attention

Expand description

Macro generated for the function crate::fast::scaled_dot_product_attention. See the function documentation for more details.