所以我有一些代码
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 似乎是一个更好的版本?