Thanks! “Good” to know I’m not the only one confused by this! 
Yeah, it also seems to be working for me, but I don’t understand why. If the output of the main Op's perform is a vector, then the output of its grad should be a matrix, unless I’m missing something.
1 Like