SequenceClsPooler

class lightautoml.ml_algo.torch_based.nn_models.SequenceClsPooler(*args, **kwargs)[source]

Bases: SequenceAbstractPooler

CLS token pooling.

forward(x, x_mask)[source]

Forward-pass.

Return type:

Tensor