(inputs, paddings) -> (encoded, paddings). Args: name: the string name of the encoder layer params. stride: To use every k-th token, set the stride to k. When stride == 0, only returns the first token of the input. When stride == 1, returns every token in the input.
(self, name, stride=1, first_n=None,
ff_hidden_dim=None, num_heads=None,
use_moe=False, layer_idx=None)
source not stored for this graph (policy: none)