mask.unsqueeze(1)
的作用是在 mask
张量的第1维度(通常是批量维度)上增加一个维度。假设原始的 mask
张量的形状是 (batch_size, seq_length)
,则 unsqueeze(1)
操作会将其变成 (batch_size, 1, seq_length)
。这一步的目的是为了后续的扩展操作做准备
unsuqeeze()操作
最新推荐文章于 2024-07-29 20:15:17 发布
mask.unsqueeze(1)
的作用是在 mask
张量的第1维度(通常是批量维度)上增加一个维度。假设原始的 mask
张量的形状是 (batch_size, seq_length)
,则 unsqueeze(1)
操作会将其变成 (batch_size, 1, seq_length)
。这一步的目的是为了后续的扩展操作做准备