把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))
相关推荐