Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tfp.experimental.nn.util.flatten_rightmost

Flatten rightmost dims.