Paddle reshape不支持0维tensor

xurqigkl  于 5个月前  发布在  其他
关注(0)|答案(2)|浏览(63)

bug描述 Describe the Bug

对带有维度为0的tensor进行reshape报错。torch可正常reshape

其他补充信息 Additional Supplementary Information

No response

svgewumm

svgewumm1#

paddle.reshape 指定 shape 中前面部分带有 0 的语义是,沿用输入的 tensor 的形状。可以参考文档中说明 paddle.reshape
所以当输入的 tensor 形状为 [2, 0] ,指定 reshape(shape=[0, 2]) 等价于 reshape(shape=[2, 2]) 。这是很早就存在的特性了,实际模型实现中也会用到这个特性。
辛苦 @zhouwei25 @zhaoyinglia 看看如何支持 0 维 tensor reshape

km0tfn4u

km0tfn4u2#

实际中会使用到的,跟torch一样的功能。

相关问题