标签: python tensorflow masking
我正在使用tf.contrib.slim来训练和评估最新的CNN。我想知道是否有一种方法可以使用tf.select或tf.where创建一个掩码来忽略某些权重在确定(甚至推论)过程中的计算。说,我想忽略值为0的权重并继续训练模型。可能吗?
tf.contrib.slim
tf.select
tf.where
任何建议/代码将不胜感激。