tensorflow 常用简单的函数收集 cond expand_dims squeeze map_fn

最新推荐文章于 2023-04-11 11:12:00 发布

TinaO-O

最新推荐文章于 2023-04-11 11:12:00 发布

阅读量234

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/u013249853/article/details/101019825

版权

本文介绍了在TensorFlow中如何使用条件函数`tf.cond`替代if else语句，以及如何通过`expand_dims`和`squeeze`进行维度变换。此外，还详细解释了如何运用`map_fn`对四维张量进行降维处理和批量操作，以便于对每个batch进行单独操作，并在完成后恢复原始形状。

摘要由CSDN通过智能技术生成

1.条件函数，替代 if else

在tensorflow里面不能使用if else对bool型tensor进行判断。所以需要使用tf.cond

https://blog.csdn.net/u013249853/article/details/101012523

x = tf.constant(2)
y = tf.constant(5)
def f1(): return tf.multiply(x, 17)
def f2(): return tf.add(y, 23)
r = tf.cond(tf.less(x, y), f1, f2)
# r is set to f1().
# Operations in f2 (e.g., tf.add) are not executed.