Function sqrt

  • Plot of sqrt and its gradient

    Calculates:

    output = sqrt(input)
    

    Gradient:

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

    Parameters

    • input: Tensor

      the input tensor of any shape

    Returns Tensor

    the output tensor

Generated using TypeDoc