description: Flatten rightmost dims.
Flatten rightmost dims.
tfp.experimental.nn.util.flatten_rightmost( ndims=3 )