A Collation
for Example<Tensor, Tensor>
types that stacks all data tensors into one tensor, and all target (label) tensors into one tensor.
More...
#include <stack.h>
Public Member Functions | |
Example | apply_batch (std::vector< Example<>> examples) override |
![]() | |
virtual OutputBatch | apply_batch (InputBatch input_batch)=0 |
Applies the transformation to the given input_batch . | |
Additional Inherited Members | |
![]() | |
using | InputBatchType = InputBatch |
using | OutputBatchType = OutputBatch |
A Collation
for Example<Tensor, Tensor>
types that stacks all data tensors into one tensor, and all target (label) tensors into one tensor.