fn is_inf(self: @Tensor<T>, detect_negative: Option<u8>, detect_positive: Option<u8>) -> Tensor<bool>;
Maps infinity to true and other values to false.
self
(@Tensor<T>
) - The input tensor.detect_negative
(Option<u8>
) - Optional Whether map negative infinity to true. Default to 1 so that negative infinity induces true.detect_positive
(Option<u8>
) - Optional Whether map positive infinity to true. Default to 1 so that positive infinity induces true.
A new Tensor<bool>
instance with entries set to true iff the input tensors corresponding element was infinity.
use core::array::{ArrayTrait, SpanTrait};
use orion::operators::tensor::{BoolTensor, TensorTrait, Tensor, U32Tensor};
fn is_inf_example() -> Tensor<bool> {
let tensor = TensorTrait::<u32>::new(
shape: array![6].span(), data: array![1, 0, NumberTrait::INF(), 8, NumberTrait::INF(), NumberTrait::INF()].span(),
);
return tensor.is_inf(detect_negative: Option::None, detect_positive: Option::None);
}
>>> [false, false, true, false, true, true]