torch (version 0.11.0)

torch_rsqrt: Rsqrt

Description

Rsqrt

Usage

torch_rsqrt(self)

Arguments

self

(Tensor) the input tensor.

rsqrt(input, out=NULL) -> Tensor

Returns a new tensor with the reciprocal of the square-root of each of the elements of input.

$${\text{out}}_i=\frac{1}{\sqrt{{\text{input}}_i}}$$

Examples

if (torch_is_installed()) {

a = torch_randn(c(4))
a
torch_rsqrt(a)
}