0、for循环 今天才发现原来for循环就能简单的遍历一个张量。虽然简单,但是很实用 1、应用场景1 目标检测中遍历bbox: import torch input = torch.rand(4, 4) for y in input: print(y) 输出结果: 2、应用场景2 遍历input中的每一张图片 import torch # B,C,H,W input = torch.rand(4, 3, 224, 224) for y in input: print(y.size())