Function prod

  • Calculates:

    output = output * input
    

    with an initial value of output = 1.0.

    Gradient:

    inputGrad = outputGrad * output / input
    

    Parameters

    • input: Tensor

      the input tensor of any shape

    • Optional dim: number | number[]
    • Optional keepdim: boolean

    Returns Tensor

    the output tensor

Generated using TypeDoc