Type alias GradientFunction

Type declaration

• • (ctx, output): (Tensor | null)[]

  • Parameters

    • ctx: GradientContext

    • output: Tensor

    Returns (Tensor | null)[]