Trait TensorAttention 

pub trait TensorAttention<T: Tensor> {
    // Required method
    fn attention(
        query: &Linear<T>,
        key: &Linear<T>,
        value: &Linear<T>,
        ctx: &mut BertContext<T>,
    ) -> Result<(), SmeltError>;
}

TODO

Required Methods

fn attention( query: &Linear<T>, key: &Linear<T>, value: &Linear<T>, ctx: &mut BertContext<T>, ) -> Result<(), SmeltError>

TODO

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<'a> TensorAttention<Tensor<'a>> for Tensor<'a>