GlobalRMSPooling1D
Applies global rms pooling for temporal data
class liqfit.modeling.pooling.GlobalRMSPooling1D
Parameters:
x (torch.Tensor): Input tensor to apply pooling on. Expected shape: (B, T, E).
attention_mask (torch.Tensor): Mask tensor to mask out the padding while applying the pooling. Expected shape: (B, T)
Using GlobalMaxPooling1D
Last updated