Template Struct Stack< TensorExample >¶
Defined in File stack.h
Page Contents
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
CollationforExample<Tensor, NoTarget>types that stacks all data tensors into one tensor.Public Functions
-
inline TensorExample apply_batch(std::vector<TensorExample> examples) override¶
-
inline TensorExample apply_batch(std::vector<TensorExample> examples) override¶
