Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.nn.util.flatten_rightmost

View source on GitHub

Flatten rightmost dims.

tfp.experimental.nn.util.flatten_rightmost(
    ndims=3
)