concepts.pdsketch.executor.expand_argument_values#

expand_argument_values(argument_values)[source]#

Expand a list of argument values to the same batch size.

Parameters:

argument_values (Sequence[TensorValue | int | str | slice | StateObjectReference]) – a list of argument values.

Returns:

the result list of argument values. All return values will have the same batch size.

Return type:

List[TensorValue]