大概的需求是这样的,我现在有两个相同形状的矩阵a,b
a:存放着所有的值,这些值的特点是有的有意义,有的没意义。
b:存放的是a矩阵中,所有位置是否有意义的标记,我把有意义的位置置为1,没意义的位置置为0,这样就得到了一个相同形状的标志矩阵b
需求:按照标记矩阵b将a中所有有意义的地方保留原来的值,而没有意义的地方都置为0
思路:使用集成好的包,numpy和torch都可以。而且坚决不用for循环,因为写for循环非常低效。
代码:
import torch
a=[[1,2,3],[5,2,3],[5,4,9]]