0
点赞
收藏
分享

微信扫一扫

TensorFlow(1.15版本),tf.map_fn速度很慢


tf.map_fn替换成tf.where实现

举例:

label_tensor = tf.map_fn(lambda x: tf.cond(x >= 1, lambda: 1, lambda: 0), label_tensor) 改成
label_tensor = tf.where(label_tensor >= 1, tf.ones_like(label_tensor), tf.zeros_like(label_tensor))


举报

相关推荐

0 条评论