It is a too long example, can you write your input tensor manually and tellme what tensor output you want like
import tensorflow as tf
tensor = tf.constant([[[1,2,3],[3,4,5]],[[6,7,8],[9,10,11]]])
print(tensor.shape)
print(tensor)