Class MultiheadAttention

• Defined in File activation.h

Inheritance Relationships

Base Type

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

Class Documentation

class MultiheadAttention : public torch::nn::ModuleHolder<MultiheadAttentionImpl>

A ModuleHolder subclass for MultiheadAttentionImpl.

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

Public Types

using Impl = MultiheadAttentionImpl