for循环遍历张量

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())