How do I modify the parameter cotangents from a custom VJP? #1868

Unanswered

NeilGirdhar asked this question in Q&A

NeilGirdhar
Feb 8, 2022

def f(x):
  ...

def f_fwd(x):
  return f(x), vjp(...)

def f_bwd(v, y_bar):
  x_bar, params_bar = v(y_bar)
  return x_bar  # How do I return params_bar here?

Replies: 0 comments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment