In validate_x_y_tensor the expected shape of y is [None] . Does it mean it can only be used for classification task?