zip迭代tensor数据、求topk的简单方法

1、zip函数可以获取可迭代数据的元素,其中包括各种list、tuple、Tensor等元素 下面是一个例子: testzip = [[1, 2], [1, 3], [2, 3], [3, 3]] print("单参数时的输出:") for tuplesin in zip(testzip): #tuplesin出来的数据是([1,2],),在3D数据训练的适合会导致其重现出来的数据
相关文章
相关标签/搜索