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