[tensorflow] tf.numpy_function() dim error 문제해결

tf.numpy_function을 사용할 때 dim 관련 오류가 날 수 있습니다.

tf.numpy_function을 이미지에 사용할 경우 reshape이 필요하기 때문에 다음과 같이 reshape을 해주어야 합니다. 

image_shape = tf.shape(image)
image = tf.numpy_function(definition(func), [image], tf.float32)
image = tf.reshape(image, image_shape)

이렇게 reshape을 해주면 dimensipn오류를 해결할 수 있습니다.