小编use*_*126的帖子

张量数组上的 python sum 与 tf.add_n

所以我有一些代码

tensors = [] //its filled with 3D float tensors
total = sum(tensors)
Run Code Online (Sandbox Code Playgroud)

如果我将最后一行更改为

total = tf.add_n(tensors)
Run Code Online (Sandbox Code Playgroud)

那么代码会产生相同的输出,但运行速度要慢得多,并且很快会导致内存不足异常。这里发生了什么?有人可以解释一下 python 内置的 sum 函数和 tf.add_n 如何分别与张量数组交互,以及为什么 python sum 似乎是一个更好的版本?

python tensorflow

3
推荐指数
1
解决办法
2163
查看次数

标签 统计

python ×1

tensorflow ×1