Function registerGradient

  • registerGradient(config): void

  • Registers a gradient function for a given kernel in the global registry, to be used during the back-propagation of that kernel.

    Parameters

    • config: GradConfig

      An object with the following properties:

      • kernelName The name of the kernel that the gradient function is for.
      • gradFunc The function to run during back-propagation.

    Returns void

Settings

Member Visibility

Theme

Generated using TypeDoc