concat_tensor_precollator ========================= .. py:module:: concat_tensor_precollator .. autoapi-nested-parse:: WARNING This file is a 1:1 duplicate from the ``tutorial`` folder. It is here to avoid installation of a tutorial as a package and keep it isolated. Classes ------- .. autoapisummary:: concat_tensor_precollator.ConcatTensorPreCollator Module Contents --------------- .. py:class:: ConcatTensorPreCollator(items, target_key, dim = 0) Bases: :py:obj:`ksuit.data.pipeline.SampleProcessor` Concatenates multiple tensors into a single tensor. :param items: A list of keys in the sample dict whose tensors should be concatenated. :param target_key: The key in the sample dict where the concatenated tensor will be stored. :param dim: The dimension along which to concatenate the tensors. Defaults to 0. .. py:attribute:: items .. py:attribute:: target_key .. py:attribute:: dim :value: 0