tf.TensorArray

 TensorArray可以看做是具有动态size功能的Tensor数组。通常都是跟while_loop或map_fn结合使用。

ta = tf.TensorArray(size, dtype, dynamic_size=True)  # dynamic_size指定数组长度可变
ta.stack(name=None)  # 将TensorArray中元素叠起来当做一个Tensor输出
ta.unstack(value, name=None)  # 可以看做是stack的反操作,输入Tensor,输出一个新的TensorArray对象,value按照其axis=0进行分割
ta.write(index, value, name=None)  # 指定index位置写入Tensor
ta.read(index, name=None)  # 读取指定index位置的Tensor

 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值