Function square

Source
pub fn square(x: &Tensor) -> Tensor