self_attention

scirs2_core::array_protocol::ml_ops

Function self_attention 

Source 
pub fn self_attention(
    queries: &dyn ArrayProtocol,
    keys: &dyn ArrayProtocol,
    values: &dyn ArrayProtocol,
    mask: Option<&dyn ArrayProtocol>,
    scale: Option<f64>,
) -> Result<Box<dyn ArrayProtocol>, OperationError>