nlp::BERTPooling Class Reference

Inheritance diagram for nlp::BERTPooling:

Collaboration diagram for nlp::BERTPooling:

Public Member Functions
	BERTPooling (dl::TransformerConf conf) noexcept
virtual dl::TensorPtr	forward (const dl::TensorPtr &input)

Detailed Description

Definition at line 48 of file bert.hpp.

Constructor & Destructor Documentation

BERTPooling()

nlp::BERTPooling::BERTPooling ( dl::TransformerConf  conf )

inlinenoexcept

Definition at line 53 of file bert.hpp.

                                                     : dense(conf.dimensions.model, conf.dimensions.model) {
            registerSubmodel("dense", dense);
        }

Member Function Documentation

forward()

virtual dl::TensorPtr nlp::BERTPooling::forward ( const dl::TensorPtr &  input )

inlinevirtual

Definition at line 57 of file bert.hpp.

{ return nullptr; }

---

The documentation for this class was generated from the following file:

• nlp/transformer/bert.hpp