|---| | [ View source on GitHub](https://github.com/tensorflow/probability/blob/v0.25.0/tensorflow_probability/python/experimental/nn/util/utils.py#L436-L455) | Reshape `op` input `x` to be a vec of `op_min_input_ndims`-rank tensors. tfp.experimental.nn.util.batchify_op( op, op_min_input_ndims, x, *other_op_args )