emmi_inference.models.pipelines.concat_tensor_precollator ========================================================= .. py:module:: emmi_inference.models.pipelines.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:: emmi_inference.models.pipelines.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