candle_nn
0.5.0
In candle_nn::rotary_emb
?
Settings
Function
candle_nn
::
rotary_emb
::
rope_i
Copy item path
source
·
[
−
]
pub fn rope_i(xs: &
Tensor
, cos: &
Tensor
, sin: &
Tensor
) ->
Result
<
Tensor
>