from scipy.fftpack import dct
import tensorflow as tf

xx=np.linspace(0,3,3)
x_dct=dct(xx,2,norm='ortho')
print(x_dct)

x= tf.placeholder(tf.float32, shape=(1,3))
x_dct=tf.spectral.dct(x,2, norm='ortho')
x_dct=tf.squeeze(x_dct)
sess=tf.Session()
sess.run(x_dct,{x:xx})