Class Flatten

• Defined in File linear.h

Inheritance Relationships

Base Type

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

Class Documentation

class Flatten : public torch::nn::ModuleHolder<FlattenImpl>

A ModuleHolder subclass for FlattenImpl.

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

Public Types

using Impl = FlattenImpl