这可能跟pytorch的版本有关
对于Tensor A,
A.size()或A.shape都是求A的size
按理来说,对Tensor起作用的操作对Veriable也能适用,但可能是版本的问题
只需要把shape改成size()就行了
管用回头记得点赞哦亲!
这可能跟pytorch的版本有关
对于Tensor A,
A.size()或A.shape都是求A的size
按理来说,对Tensor起作用的操作对Veriable也能适用,但可能是版本的问题
只需要把shape改成size()就行了
管用回头记得点赞哦亲!