Class Unflatten

• Defined in File linear.h

Inheritance Relationships

Base Type

• public torch::nn::ModuleHolder< UnflattenImpl > (Template Class ModuleHolder)

Class Documentation

class Unflatten : public torch::nn::ModuleHolder<UnflattenImpl>

A ModuleHolder subclass for UnflattenImpl.

See the documentation for UnflattenImpl class to learn what methods it provides, and examples of how to use Unflatten with torch::nn::UnflattenOptions. See the documentation for ModuleHolder to learn about PyTorch’s module storage semantics.

Public Types

using Impl = UnflattenImpl