h_img, h_crop等都是数据的size,应该都是二维三维的,不能直接跟后面的1做加减, 你可以尝试一下将 1 扩展成等维度的数据
报错意思就是不支持torch.size和int类型进行运算,你把1改成torch.ones((宽度,高度))试