nip.utils.nested_array_dict.stack_nested_array_dicts#
- nip.utils.nested_array_dict.stack_nested_array_dicts(nds: Sequence[NestedArrayDict], dim=0) NestedArrayDict [source]#
Stack a sequence of NestedArrayDicts along a new dimension.
- Parameters:
nds (Sequence[NestedArrayDict]) – The NestedArrayDicts to stack. All NestedArrayDicts must have the same keys, shapes and batch sizes.
dim (int, default=0) – The dimension to stack along. This must be a non-negative integer at most equal to the number of batch dimensions.
- Returns:
stacked (NestedArrayDict) – The NestedArrayDict with the stacked arrays. The batch size of the new NestedArrayDict is the batch size of the input NestedArrayDicts with the new dimension added.