Tensorflow 로우 와이즈 1D 회선

Question

I이 용지 거치지 하였다 CNN for Sentence Modelling가

용지에 다음 선을 따라 컨벌루션 설명 같다 : 행렬 입력 와

를 I 크기 [D, m] 크기의 W 가중치 행렬 [D, S]

컨벌루션는 입력 행렬의 각 행은 가중치 행렬의 각 행 (기본적으로는 행 방향으로 1 차원 컨볼 루션)으로 선상처럼 수행된다. 출력은 적절한 패딩 또는 [d, m-s + 1] 패딩없이 크기 [d, m]가됩니다.

나의 이해는 이것과 단순히 1D 컨볼 루션을하는 것의 유일한 차이점은 입력 행렬의 각 행이 그 자체의 가중치 세트를 가질 것이라는 점입니다.

Tensorflow에서 이러한 회선 레이어를 구현할 수 있습니까?

Answer 1

이렇게 할 수 있습니다.

I = tf.reshape(I, [1,d,1,m]) 
W = tf.reshape(W, [1,d,s,1) 
strides = [1,stride,1,1] 
output = tf.nn.depthwise_conv2d(I, W, strides, padding='SAME') 

나는 이것이 당신이 찾고있는 것이라고 생각합니다. 입력에 걸쳐 각 필터 행을 적용하고 컨벌루션 된 결과를 반환합니다.