Created by: iclementine

fix kron_op: when only one input needs gradient

Kron op wroks like matmul. It has two inputs, X and Y. So we need to check their gradients altogether, and separately.