tf.squeeze를 tf.reshape로 대체하십시오.

Question

tensorflow를 사용하여 모바일 네트워크를 학습해야합니다. tf.squeeze 계층은 지원되지 않습니다. tf.reshape로 바꿀 수 있습니까?

는 동작이다 그물 모양 [50,1,1,1000]을 갖는다

tf.reshape(net, [50,1000], name='reshape') 

: 같이

tf.squeeze(net, [1, 2], name='squeeze') 

동일.

Answer 1

왜 tf.squeeze은 지원되지 않습니까? 텐서에서 1 차원 축을 제거하려면 tf.squeeze이 올바른 연산입니다. 하지만 tf.reshape을 사용하여 원하는 작업을 수행 할 수도 있지만 tf.squeeze을 사용하도록 제안합니다.