技术栈
xla
money0534
3 天前
pytorch
·
线性代数
·
矩阵
·
transformers
·
xla
XLA中生成Causal Mask上三角-inf矩阵
transformers生成CausalAttentionMask的上三角-inf矩阵: 参考transformers源码