Versioned name: Rsqrt-1

Category: Arithmetic

Short description: Rsqrt performs element-wise reciprocal of square root operation with given tensor.


  • 1: A tensor of type T. Required.

    • Type: T


  • 1: The result of element-wise Rsqrt operation. Required.

    • Type: T


  • T: f32, f16, bf16.

  • Note: Inputs and outputs have the same data type denoted by T. For example, if input is f32 tensor, then all other tensors have f32 data type.

Rsqrt does the following with the input tensor a:

\[a_{i} = 1 / sqrt(a_{i})\]