xla

pytorch·线性代数·矩阵·transformers·xla

XLA中生成Causal Mask上三角-inf矩阵transformers生成CausalAttentionMask的上三角-inf矩阵：参考transformers源码