FirstTokenPooling1D

Applies first token pooling for temporal data

class liqfit.modeling.pooling.FirstTokenPooling1D

forward(x: torch.Tensor)

Parameters:

• x (torch.Tensor): Input tensor to apply pooling on. Expected shape: (B, T, E).

Using FirstTokenPooling1D

from liqfit.modeling.pooling import FirstTokenPooling1D
import torch

x = torch.randn((1, 10, 20))
pooler = FirstTokenPooling1D()
out = pooler(x)