masked_fill

码农研究僧25 天前
人工智能·pytorch·python·masked_fill
详细分析Pytorch中的masked_fill基本知识（附Demo）基本的原理知识如下：输入张量和掩码： masked_fill 接受两个主要参数：一个输入张量和一个布尔掩码 掩码的形状必须与输入张量相同，True 表示需要填充的位置，False 表示保持原值