pulsatrix/api/text__collate_8hpp_source.html

#pragma once


#include "pulsatrix/collate.hpp"


namespace pulsatrix {


[[nodiscard]] CollateFn PadCollate(float pad_index = 0.0f);


}  // namespace pulsatrix

collate.hpp
Batch assembly – Batch, CollateFn, DefaultCollate.

pulsatrix
Definition acquisition_functions.hpp:16

pulsatrix::CollateFn
std::function< Batch(std::vector< Sample >, DeviceBackend *)> CollateFn
A function assembling a list of Samples into one Batch – pulsatrix's analogue of PyTorch's collate_fn...
Definition collate.hpp:30

pulsatrix::PadCollate
CollateFn PadCollate(float pad_index=0.0f)
Builds a CollateFn that right-pads variable-length token sequences (sample.fields[0],...