concepts.dsl.tensor_value_utils.expand_argument_values#

expand_argument_values(argument_values, handle_wildcard=False)[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.

  • handle_wildcard (bool) – whether to handle the wildcard variable ‘??’. If set to True, the function will return the original argument values without expanding them if any of the argument values contains the wildcard variable.

Returns:

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

Return type:

List[TensorValue]