torch.nn.TransformerEncoderLayerEncodeOptions
export interface TransformerEncoderLayerEncodeOptions {
/** Optional attention mask */
src_mask?: Tensor;
/** Optional padding mask for variable-length sequences */
src_key_padding_mask?: Tensor;
/** If true, applies causal attention mask */
is_causal?: boolean;
}src_mask(Tensor)optional- – Optional attention mask
src_key_padding_mask(Tensor)optional- – Optional padding mask for variable-length sequences
is_causal(boolean)optional- – If true, applies causal attention mask
Options for TransformerEncoderLayer.encode