写 pytorch 代码时,想查看某个 tensor 的某个维度,一开始用成了 .shape
if test_image.shape[1] == 3:
......
结果报错 :
RuntimeError: Boolean value of Tensor with more than one value is ambiguous
这是因为 tensor 不能用 .shape,而应该用 .size
于是我就写成了 .size,
if test_image.size[1] == 3:
......
结果报错:
TypeError: 'builtin_function_or_method' object is unsubscriptable
这是因为括号 [ ] 写错了,应该用 ():
if test_image.size(1) == 3:
......
解决问题。