Method cosine

cake-core/src/models/common/cache.rs:140–142

Return the cached cosine value for the given position and sequence length. When `device` differs from the cache's own device, the result is copied to that device (enables multi-GPU workers).

(&self, index_pos: usize, seq_len: usize, device: &Device)

    /// When `device` differs from the cache's own device, the result is copied
    /// to that device (enables multi-GPU workers).
    pub fn cosine(&self, index_pos: usize, seq_len: usize, device: &Device) -> Result<Tensor> {
        self.cos.narrow(0, index_pos, seq_len)?.to_device(device)
    }

    /// Return the cached sine value for the given position and sequence length.
    pub fn sine(&self, index_pos: usize, seq_len: usize, device: &Device) -> Result<Tensor> {