哪个张量流操作会触发内存分配?

问题描述 投票:0回答:1

阅读TensorArray类的文档后,我遇到了这一行:

如果为True,TensorArray将与Tensor首次写入时使用的设备位于同一设备上(写入操作包括写入,取消堆栈和拆分)。如果为False,则TensorArray将放置在设备上,该设备由初始化期间可用的设备上下文确定。

这使我想知道是否只有某些操作会触发实际的内存分配和读取/写入,以及是否有这些操作的列表。

我也想知道这一点,因为创建张量临时存在很大的限制,但是由于OOM错误,这些张量会以批处理方式进行处理,因此从不需要程序实际保存张量的所有数据-类似于数据集API允许按需加载所有内容时<< Infinitely >>大型数据集的方式。

在阅读TensorArray类的文档时,我遇到了以下这一行:如果为True,则TensorArray将与首次写入(写...]时使用的Tensor位于同一设备上。) [

从Tensorflow-GPU角度来看:没有列表。

python tensorflow machine-learning memory out-of-memory
1个回答
0
投票
没有XLA
© www.soinside.com 2019 - 2024. All rights reserved.