我想稀疏卷积内核,所以我需要在训练过程中将内核中的一些值设置为零值.张量流中是否有一些api可以帮助我实现我的想法,将张量中的某些值设置为零？
gde*_*lab 10
您可以使用tf.boolean_mask(original_tensor, mask)仅保留所需的值(您将删除其他值而不是将它们设置为0).
要保持初始形状并且在某些地方只有零,您可以执行以下操作:
new_tensor = tf.multiply(original_tensor, tf.cast(mask, original_tensor.type()))
对于您的示例,您可以使用sthg构建掩码,如:
mask = tf.less(original_tensor, 0.0001 * tf.ones_like(original_tensor))
| 归档时间： |
|
| 查看次数： |
6415 次 |
| 最近记录： |