- class mindspore_rl.utils.TensorsQueue(dtype, shapes, size=0, name='TQ')[源代码]
用来存Tensor列表的队列。
警告
这是一个实验特性,未来有可能被修改或删除。
- 参数:
dtype (mindspore.dtype) - Tensor队列的数据类型。每个Tensor需要相同的类型。
shapes (tuple[int64]) - Tensor队列中每个Tensor的shapes。
size (int) - Tensor队列的大小。默认:0。
name (str) - Tensor队列的名字。默认:”TQ”。
- 异常:
TypeError - dtype 不是 MindSpore 数字类型.
ValueError - size 小于0.
ValueError - shapes 的长度小于1.
- 支持平台:
GPU
CPU
样例:
>>> import mindspore as ms >>> from mindspore import Tensor >>> from mindspore_rl.utils import TensorsQueue >>> data1 = Tensor([[0, 1], [1, 2]], dtype=ms.float32) >>> data2 = Tensor([1], dtype=ms.float32) >>> tq = TensorsQueue(dtype=ms.float32, shapes=((2, 2), (1,)), size=5) >>> tq.put((data1, data2)) >>> ans = tq.pop()
- close()[源代码]
关闭动态数组。
警告
一旦关闭了Tensor队列,每个属于该Tensor队列的方法都将失效。所有该队列中的资源也将被清除。如果该队列还将在别的地方使用,如下一个循环,请用 clear 代替。
- 返回:
True。