Fixed TensorInv infer_shape.