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 即可
ValueError: Shape must be rank 1 but is rank 2 for 'reverse_sequence'
最新推荐文章于 2021-06-20 13:40:04 发布