e_w = gradients[i] * self.rho / (grad_norm + 1e-12) TypeError: unsupported operand type(s) for *: 'IndexedSlices' and 'float' How to deal with it?