jax-js-nonconsuming (Fork)
    Preparing search index...

    Function scaleByLearningRate

    • Scale by the (negative) learning rate (either as scalar or as schedule).

      Parameters

      • learningRate: number | ((count: Array) => Array)
      • flipSign: boolean = true

      Returns GradientTransformation