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: ‘ReverseSequence’)with input shapes: [3,3], [1,3].
查了半天也没明白是错在哪,回过头来细想才发现是自己理解错误,
reverse_sequence的参数seq_lengths的维度dim只能为1维,
按照param文件中所写seq_dims:1,3 就是2维 ,
修改seq_dims: 3 即可
ValueError: Shape must be rank 1 but is rank 2 for 'reverse_sequence'
最新推荐文章于 2021-06-20 13:40:04 发布