Typedef torch::data::transforms::Collation

• Defined in File collate.h

Typedef Documentation

template<typename T, typename BatchType = std::vector<T>>
using torch::data::transforms::Collation = BatchTransform<BatchType, T>

A Collation is a transform that reduces a batch into a single value.

The result is a BatchDataset that has the type of the single value as its BatchType.