pub mod linear;
pub mod quadratic;
use ::burn::tensor::{Tensor, backend::Backend};
const EPSILON: f32 = 1e-6;
type Values<B> = Tensor<B, 1>;
fn safe_divisor<B: Backend>(tensor: Values<B>) -> Values<B> {
tensor
.clone()
.mask_fill(tensor.abs().lower_equal_elem(EPSILON), 1.0)
}