How can I output a gradient in vector format in Op.grad instance?

Questions v5
4

Perhaps have a look at the gradient of some vector output Ops in Aesara, like Softmax? aesara/basic.py at 9f176da71d41635e8854fd601fd6a68102b0c6e5 · aesara-devs/aesara · GitHub

1 Like