Shortcuts

Template Struct Stack< TensorExample >

Inheritance Relationships

Base Type

Struct Documentation

template<>
struct torch::data::transforms::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

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources