- class mindspore_rl.utils.TensorArray(dtype, element_shape, dynamic_size=True, size=0, name='TA')[源代码]
用来存Tensor的动态数组。
警告
这是一个实验特性,未来有可能被修改或删除。
- 参数:
dtype (mindspore.dtype) - 动态数组的数据类型。
element_shape (tuple(int)) - 动态数组中每个Tensor的shape。
dynamic_size (bool) - 如果是true,则该数组可以动态增长,否则为固定大小。默认:True。
size (int) - 如果 dynamic_size=False , 则 size 表示该数组的最大容量。
name (str) - 动态数组的名字。默认:”TA”。
- 支持平台:
GPU
CPU
样例:
>>> import mindspore >>> from mindspore_rl.utils import TensorArray >>> ta = TensorArray(mindspore.int64, ()) >>> ta.write(0, 1) >>> ta.write(1, 2) >>> ans = ta.read(1) >>> print(ans) 2 >>> s = ta.stack() >>> print(s) [1 2] >>> ta.clear() >>> ta.write(0, 3) >>> ans = ta.read(0) >>> print(ans) 3 >>> ta.close()