把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))
把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))
7560
1650
1273

被折叠的 条评论
为什么被折叠?
>