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數據訓練的適合會導致其重現出來的數據
相關文章
相關標籤/搜索