pub fn sign(input: &Tensor) -> Result<Tensor>Expand description
Compute elementwise signs.
ยงExamples
use tenferro_cpu::sign;
use tenferro_tensor::Tensor;
let input = Tensor::from_vec_col_major(vec![3], vec![-2.0_f64, 0.0, 3.0])?;
let out = sign(&input)?;
assert_eq!(out.as_slice::<f64>().unwrap(), &[-1.0, 0.0, 1.0]);