如何检查两个张量的形状不兼容？

发布于 2025-01-27 23:10:19 字数 321 浏览 1 评论 0原文

是否有一种简单的方法来检查两个张量是否具有不兼容的形状（即，一个不可能的形状无法广播到另一个形状）？诸如

RuntimeError: The size of tensor a (?) must match the size of tensor b (?) at non-singleton dimension ?

我可以使用t1.size（）[2]！= t2.size（）[2]之类的错误（即，我知道何时以及如何发生）更复杂的场景），但是我如何习惯地说呢？

原文

Is there a simple way to check if two tensors have incompatible shapes (i.e., one can not be broadcast to the shape of the other) before multiplying them? Errors such as

RuntimeError: The size of tensor a (?) must match the size of tensor b (?) at non-singleton dimension ?

I could use t1.size()[2] != t2.size()[2] in what I do (i.e., I know when and how that happens, but there could be more complicated scenarios), but how do I say it idiomatically?

分享到QQ

分享到微博