Function rsqrt

  • Plot of rsqrt and its gradient

    Calculates:

    output = 1.0 / sqrt(input)

    Gradient:

    inputGrad = -outputGrad / (2.0 * sqrt(input) * input)

    Parameters

    • input: Tensor

      the input tensor of any shape

    Returns Tensor

    the output tensor

Settings

Member Visibility

Theme

Generated using TypeDoc