This method looks wrong. It looks like you start fprop'ing from `state` as if `state` were the input, but it is in fact the output. Have you tested this? If so I don't understand why you're not getting size mismatch errors.