When a gradient is zero because the variable has no influence on the result, now return a variable with same shape instead of a 0 scalar, so that we can rely on the shape being unchanged