ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tfp.experimental.nn.util.batchify_op

Reshape op input x to be a vec of op_min_input_ndims-rank tensors.