0) src_mask = src_mask.float().masked_fill(src_mask == 0, float('-inf')).masked_fill(src_mask == 1, float(0.0)) src_mask = src_mask.cuda() ...
確定! 回上一頁