torch_rsqrt function

Rsqrt

Rsqrt

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.

\mboxouti=1\mboxinputi \mbox{out}_{i} = \frac{1}{\sqrt{\mbox{input}_{i}}}

Examples

if (torch_is_installed()) { a = torch_randn(c(4)) a torch_rsqrt(a) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14