阅读TensorArray类的文档后,我遇到了这一行:
如果为True,TensorArray将与Tensor首次写入时使用的设备位于同一设备上(写入操作包括写入,取消堆栈和拆分)。如果为False,则TensorArray将放置在设备上,该设备由初始化期间可用的设备上下文确定。
这使我想知道是否只有某些操作会触发实际的内存分配和读取/写入,以及是否有这些操作的列表。
我也想知道这一点,因为创建张量临时存在很大的限制,但是由于OOM错误,这些张量会以批处理方式进行处理,因此从不需要程序实际保存张量的所有数据-类似于数据集API允许按需加载所有内容时<< Infinitely >>大型数据集的方式。
在阅读TensorArray类的文档时,我遇到了以下这一行:如果为True,则TensorArray将与首次写入(写...]时使用的Tensor位于同一设备上。) [从Tensorflow-GPU角度来看:没有列表。