Comment on page
nn.sigmoid
fn sigmoid(tensor: @Tensor<T>) -> Tensor<T>;
Applies the Sigmoid function to an n-dimensional input tensor rescaling them so that the elements of the n-dimensional output Tensor lie in the range [0,1].
tensor
(@Tensor<T>
) - The input tensor.
A Tensor of fixed point numbers with the same shape than the input Tensor.
Constrain input and output types to fixed point tensors.
use array::{ArrayTrait, SpanTrait};
use orion::operators::tensor::{TensorTrait, Tensor, FP8x23};
use orion::operators::nn::{NNTrait, FP8x23NN};
use orion::numbers::{FP8x23, FixedTrait};
fn sigmoid_example() -> Tensor<FP8x23> {
let tensor = TensorTrait::<FP8x23>::new(
shape: array![2, 2].span(),
data: array![
FixedTrait::new(0, false),
FixedTrait::new(1, false),
FixedTrait::new(2, false),
FixedTrait::new(3, false),
]
.span(),
);
return NNTrait::sigmoid(@tensor);
}
>>> [[4194304,6132564],[7388661,7990771]]
// The fixed point representation of
// [[0.5, 0.7310586],[0.88079703, 0.95257413]]
Last modified 2mo ago