Template Struct Stack< TensorExample >

• Defined in File stack.h

Inheritance Relationships

Base Type

• public torch::data::transforms::BatchTransform< Example< Tensor, example::NoTarget > > (Template Class BatchTransform)

Struct Documentation

template<>
struct Stack<TensorExample> : public torch::data::transforms::BatchTransform<Example<Tensor, example::NoTarget>>

A Collation for Example<Tensor, NoTarget> types that stacks all data tensors into one tensor.

Public Functions

inline TensorExample apply_batch(std::vector<TensorExample> examples) override