Square

torch_square(self)

Arguments

self

(Tensor) the input tensor.

square(input, out=NULL) -> Tensor

Returns a new tensor with the square of the elements of input.

Examples

if (torch_is_installed()) { a = torch_randn(c(4)) a torch_square(a) }
#> torch_tensor #> 0.1858 #> 0.0154 #> 0.1483 #> 0.8034 #> [ CPUFloatType{4} ]