use core::array::{ArrayTrait, SpanTrait};
use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
fn squeeze_example() -> Tensor<u32> {
let tensor = TensorTrait::<u32>::new(
shape: array![1, 2, 1, 2, 1].span(),
data: array![1, 1, 1, 1].span(),
);
return tensor.squeeze(axes: Option::None(());
}
>>> [[1 1]
[1 1]]