noether.data.pipeline.sample_processors.concat_tensor¶

Classes¶

Concatenates multiple tensors into a single tensor.

class noether.data.pipeline.sample_processors.concat_tensor.ConcatTensorSampleProcessor(items, target_key, dim=0)¶

Bases: noether.data.pipeline.SampleProcessor

Concatenates multiple tensors into a single tensor.

Parameters:

items (list[str]) – A list of keys in the sample dict whose tensors should be concatenated.
target_key (str) – The key in the sample dict where the concatenated tensor will be stored.
dim (int) – The dimension along which to concatenate the tensors. Defaults to 0.