torch.nn.TransformerRunOptions
export interface TransformerRunOptions {
/** Optional attention mask for encoder */
src_mask?: Tensor;
/** Optional attention mask for decoder self-attention */
tgt_mask?: Tensor;
/** Optional attention mask for decoder cross-attention */
memory_mask?: Tensor;
/** Optional padding mask for source sequence */
src_key_padding_mask?: Tensor;
/** Optional padding mask for target sequence */
tgt_key_padding_mask?: Tensor;
/** Optional padding mask for memory */
memory_key_padding_mask?: Tensor;
/** If true, applies causal mask to encoder */
src_is_causal?: boolean;
/** If true, applies causal mask to decoder self-attention */
tgt_is_causal?: boolean;
}src_mask(Tensor)optional- – Optional attention mask for encoder
tgt_mask(Tensor)optional- – Optional attention mask for decoder self-attention
memory_mask(Tensor)optional- – Optional attention mask for decoder cross-attention
src_key_padding_mask(Tensor)optional- – Optional padding mask for source sequence
tgt_key_padding_mask(Tensor)optional- – Optional padding mask for target sequence
memory_key_padding_mask(Tensor)optional- – Optional padding mask for memory
src_is_causal(boolean)optional- – If true, applies causal mask to encoder
tgt_is_causal(boolean)optional- – If true, applies causal mask to decoder self-attention
Options for Transformer.run