所以,我想屏蔽SparseTensor
的整行。用tf.boolean_mask
很容易做到这一点,但是SparseTensor
没有等价的方法。目前,我有可能只检查SparseTensor.indices
中的所有索引,然后过滤掉所有不是屏蔽行的索引,例如:
masked_indices = list(filter(lambda index: masked_rows[index[0]], indices))
其中masked\u rows是一个1D数组,表示该索引处的行是否被屏蔽。你知道吗
然而,这真的很慢,因为我的SparseTensor相当大(它有90k个索引,但会越来越大)。在我对过滤的索引应用SparseTensor.mask
之前,在单个数据点上花费了相当多的时间。这种方法的另一个缺陷是,它实际上也没有删除行(尽管在我的例子中,一个全零的行也一样好)。你知道吗
有没有更好的方法来屏蔽一个行稀疏传感器,还是这是最好的方法?你知道吗
你可以这样做:
相关问题 更多 >
编程相关推荐