2017-03-01 5 views
0

저는이 문제를 꽤 오랫동안 풀려고하고 있지만 여전히 해결 방법을 찾지 못하고 있습니다. 아마도 누군가 여기서 나를 도울 수 있습니다. 나는 신경 네트워크에 다음 AAABBB 입력 테이블이 있습니다AAABBB 테이블을 횃불의 텐서 (tensors) 중첩 된 ABABAB 테이블로 변환하십시오.

{ 
    1 : 
    { 
     1 : DoubleTensor - size: 32x200 
     2 : DoubleTensor - size: 32x200 
     3 : DoubleTensor - size: 32x200 
    } 
    2 : 
    { 
     1 : DoubleTensor - size: 32x54 
     2 : DoubleTensor - size: 32x54 
     3 : DoubleTensor - size: 32x54 
    } 
} 

위의 표는 전처리 한 후 중첩ababab와 입력 테이블로 변환해야됩니다 :

{ 
    1 : 
    { 
     1 : DoubleTensor - size: 32x200 
     2 : DoubleTensor - size: 32x54 
    } 
    2 : 
    { 
     1 : DoubleTensor - size: 32x200 
     2 : DoubleTensor - size: 32x54 
    } 
    3 : 
    { 
     1 : DoubleTensor - size: 32x200 
     2 : DoubleTensor - size: 32x54   
    } 
} 

방법 변환 할 수 있습니까 AAABBB 테이블을 ABABAB중첩이내 네트워크에서 Torch table layers을 사용하고 계십니까?

답변

0

dpnn 패키지에는 정확히 Container이 포함되어 있습니다. ZipTable 테이블 테이블을 테이블 테이블로 압축합니다. 여기

는 테이블을 중첩 ababab와에 AAABBB 테이블을 변환하는 방법의 일 예이다.

require 'dpnn' 

aaa = torch.DoubleTensor(3,32,200) 
bbb = torch.DoubleTensor(3,32,54) 

model = nn.Sequential() 
par = nn.ParallelTable() 
par:add(nn.SplitTable(1)) 
par:add(nn.SplitTable(1)) 
model:add(par) 
model:add(nn.ZipTable()) 
model:forward({aaa,bbb})