main
Ask or search…
K
Links

tensor.erf

tensor.erf

fn erf(self: @Tensor<T>) -> Tensor<T>;
Computes the mean of the input tensor's elements along the provided axes.

Returns

A new Tensor<T> of the same shape as the input tensor with the the error function of the input tensor computed element-wise.

Type Constraints

Constrain input and output types to fixed point tensors.

Examples

use core::array::{ArrayTrait, SpanTrait};
use orion::operators::tensor::{TensorTrait, Tensor, FP16x16Tensor};
use orion::numbers::{FixedTrait, FP16x16};
fn erf_example() -> Tensor<FP16x16> {
// The erf inputs is [1.0, 0.134, 0.520, 2.0, 3.5, 5.164]
let tensor = TensorTrait::<FP16x16>::new(
shape: array![6].span(),
data: array![
FixedTrait::new_unscaled(65536, false),
FixedTrait::new_unscaled(8832, false),
FixedTrait::new_unscaled(34079, false),
FixedTrait::new_unscaled(131072, false),
FixedTrait::new_unscaled(229376, false),
FixedTrait::new_unscaled(338428, false),
]
.span(),
);
return tensor.erf();
}
>>> [55227,9560,35252,65229,65536,65536]