pyvene.models.basic_utils.get_batch_size

Contents

pyvene.models.basic_utils.get_batch_size#

get_batch_size(model_input)[source]#

Get batch size based on the input