Update DiagonalSubtensor.grad to don't return None.
Update IncDiagonalSubtensor.grad to don't return None.
pep8