(self, activation: PoolerActivation)
| 233 | class PoolerHead(nn.Layer): |
| 234 | |
| 235 | def __init__(self, activation: PoolerActivation) -> None: |
| 236 | super().__init__() |
| 237 | self.activation = activation |
| 238 | |
| 239 | def forward(self, pooled_data: Union[list[paddle.Tensor], paddle.Tensor], pooling_metadata: PoolingMetadata): |
| 240 |