Python3中assert语句后追加(x.shape, y.shape)参数的作用是什么？

逗号后内容的定义 #

Python 的 assert 语法结构为：assert 判断条件, 自定义异常提示

• 逗号前的x.shape == y.shape是纯判断逻辑：当两个数组/张量的形状完全匹配时，程序跳过这行代码继续运行；当形状不匹配时，立刻中断程序并抛出AssertionError。
• 逗号后的(x.shape, y.shape)是异常触发时的附加提示内容，它是一个元组结构，完全不参与前面的真假判断。

设置这段提示的实际作用 #

• 如果省略这部分提示，形状不匹配时控制台只会输出干巴巴的AssertionError，你无法快速知道是哪两个变量维度不一致、各自的维度具体是多少，需要额外加断点、加打印语句定位问题。
• 加上这段元组提示后，一旦校验失败，控制台会直接输出类似AssertionError: (torch.Size([2, 32]), torch.Size([4, 32]))的报错内容，你一眼就能看清两个变量的实际形状差异，不需要额外调试就能定位维度不匹配的具体问题。

常见误区：不要误以为元组内容会参与判断。不管提示项是什么类型的内容，只要前面的形状相等判断为假，就会直接抛出异常并把对应提示带出来，提示项本身永远不会被当成判断条件求值。

内容的提问来源于stack exchange，提问作者Megafreak731