我有一個熊貓數(shù)據(jù)框200列乘2500行,我把它做成張量tensor = torch.tensor(df.values)tensor.size() => ([2500,200])我分塊列舉了list=[]for i,chunk in enumerate(tensor.chunk(100,dim=0)) chunk.size =>([25,200]) output = hiddenlayer(chunks) output.size() => ([25,1]) list += output塊被饋送通過一些層,并輸出為1個特征張量。所以現(xiàn)在我有一個100張量的列表,每個張量有25個1、100x25x1的塊所以我stacked = torch.stack(list, 1).squeeze(2)stacked.size()=([25,100])我已經(jīng)嘗試過堆疊和壓縮,但是我似乎無法回到我想要的([2500,1])。我想念什么嗎?如果您能迅速幫助我了解堆疊和壓縮的功能以及為什么它對我不起作用,那么您將永遠背負債務!謝謝
擠壓后的pytorch .stack最終形狀
慕桂英3389331
2021-05-21 18:15:57