main
Ask or search…
K
Links
Comment on page

tensor.cos

#tensor.cos
fn cos(self: @Tensor<T>) -> Tensor<T>;
Computes the cosine of all elements of the input tensor.

Args

  • self(@Tensor<T>) - The input tensor.

Returns

A new Tensor<T> of the same shape as the input tensor with the cosine value of all elements in the input tensor.

Type Constraints

Constrain input and output types to fixed point tensors.

Example

use array::{ArrayTrait, SpanTrait};
use orion::operators::tensor::{TensorTrait, Tensor, FP8x23Tensor};
use orion::numbers::{FP8x23, FixedTrait};
fn cos_example() -> Tensor<FP8x23> {
let tensor = TensorTrait::<FP8x23>::new(
shape: array![3].span(),
data: array![
FixedTrait::new_unscaled(0, false),
FixedTrait::new_unscaled(1, false),
FixedTrait::new_unscaled(2, false)
]
.span(),
);
return tensor.cos();
}
>>> [8388608,4532384,-3490893]
// The fixed point representation of
// [1, 0.5403...,-0.4161]