enumerate（）函数用在tensor数组上

 enumerate（）函数用在tensor数组上，会让数组按第0维展开

import torch

pred=torch.arange(0, 20, 2).reshape([2,5])

for i, det in enumerate(pred):
    print(f'i:{i}')
    print(f'det:{det}')

结果：

enumerate()正常用法

enumerate() 函数会同时列出数据和数据下标，一般用在 for 循环当中。

seasons = ['Spring', 'Summer', 'Fall', 'Winter']
aaa=list(enumerate(seasons))
print(aaa)

结果 

[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]