내가 3D 텐서 <strong>(sequence_length, BATCH_SIZE, word_dim)</strong>이 어떻게 tensorflow

Question

에서 특정 차원에 matmul 작업을 수행하는, 내가 (에 텐서 변경할 수 있도록 "word_dim"차원와 matmul 작업을 할 필요가 sequence_length, batch_size, hidden_dim). matmul 연산은 2D 텐서에서만 사용할 수 있습니다. 그리고 "batch_size"때문에 3D 텐서를 2D로 변경할 수 없습니다. 어떻게해야합니까?

Answer 1

내가 처음 로 당신의 텐서를 바꿀 것 (sequence_length * BATCH_SIZE, word_dim), 얻을 수있는 matmul을 (sequence_length * BATCH_SIZE, hidden_dim), 다음 얻기 위해 다시 바꿀 (sequence_length, BATCH_SIZE, hidden_dim). reshape()와 관련된 복사 작업은 없으며, 곱하기 하나의 행렬 만 있으면 batch_size 행렬에 개별적으로 곱하는 것과 같습니다.