pytorch利用類似掩碼的功能把一些值置為0

在pytorch中，簡單利用float（）生成掩碼，然后過濾掉不想用的值，保存維度不變。