2016-07-13 2 views
0

토치에서 일괄 처리에 사용할 입력의 치수는 어느 것입니까?토치 모듈에 공급하기 전에 입력을 일괄 처리하는 방법은 무엇입니까?

나는 훈련을위한 예제가 1000 개 있고 각 훈련 예는 10 * 5 차원입니다. 자,이 데이터를 시퀀서에 100 개의 예제로 묶어주고 싶습니다.

어떻게 입력을 구성해야합니까? 입력의 각 배치의 차원이 100 * 10 * 5 (일괄 처리에 사용 된 첫 번째 차원) 또는 10 * 100 * 5 (일괄 처리에 두 번째 차원 사용)이어야합니까?

다음 규칙을 설명하는 관련 문서에 대한 링크를 참조하십시오.

용기 및 모듈에 대한 규정이 변경됩니까?

답변

0

일반적으로 크기가 100*10*5 인 텐서입니다. 이미지라면 채널 수를 고려하여 batchSize*channels*width*height이되어야 할 수도 있습니다. 이렇게하면 데이터에 쉽게 액세스 할 수 있습니다. 데이터를 검색하려면 inputs[{i}]을 수행하면됩니다. 레이블을 저장할 다른 Tensor를 만드는 것을 고려하십시오 (레이블이 지정된 데이터를 사용하는 경우). 여기에서 예제를 찾을 수 있습니다. https://github.com/torch/tutorials/blob/master/2_supervised/4_train.lua#L131

내가 튜토리얼을 좀보고 당신을 권하고 싶습니다, 거기 당신은 사실은 내가이를보고 난 후에 혼란 스러워요, 데이터가 네트워크 https://github.com/torch/tutorials

+1

감사 마누엘를 공급하기 전에 "준비"해야하는 방법을 볼 수 있습니다 - https://github.com/Element-Research/rnn#rnn.Sequencer 두 번째 차원은 기본적으로 배치에 사용됩니다. 나는 다른 용기가 다른 관례를 사용하고 있다고 생각한다. –

+0

@VikramGupta 맞습니다! 이 경우에는 시퀀서 컨테이너가 다른 입력 모양을 필요로하는 것처럼 보입니다. –

관련 문제