Typedef torch::data::transforms::Collate

• Defined in File collate.h

Typedef Documentation

template<typename T, typename BatchType = std::vector<T>>
using torch::data::transforms::Collate = BatchLambda<BatchType, T>

A Collate allows passing a custom function to reduce/collate a batch into a single value.

It’s effectively the lambda version of Collation, which you could subclass and override operator() to achieve the same.