ValueError: Shape must be rank 1 but is rank 2 for 'reverse_sequence'

python tf_reverse_sequence.py reverse_001_int8.param reverse_001_output_expect_int8.bin
执行脚本生成数据bin文件时,总是报错;
ValueError: Shape must be rank 1 but is rank 2 for ‘reverse_sequence’ (op: ‘Reverse_sequence’) with input shapes :[3,3] , [1,3].
seq_lengths = [1,2,3]
而param文件如下图所示:
在这里插入图片描述
查了半天也没明白错在哪,回过头来想明白是理解错误;
reverse的seq_lengths的维度dim只能是1维,而这里seq_dims:1,3是2维了;
修改seq_dims:3 即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值