Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

tfp.experimental.nn.util.flatten_rightmost

Flatten rightmost dims.