poolio

#Instructions for installing OpenBLAS on a terminal.com instance. Mostly folows: http://stackoverflow.com/questions/11443302/compiling-numpy-with-openblas-integration

##Run the following:

sudo dpkg -r python-numpy

sudo apt-get install gfortran
sudo pip install cython

poolio

def _phase_shift(I, r):
    bsize, a, b, c = I.get_shape().as_list()
    X = tf.reshape(I, (bsize, a, b, r, r))
    X = tf.transpose(X, (0, 1, 2, 4, 3))  # bsize, a, b, 1, 1
    X = tf.split(1, a, X)  # a, [bsize, b, r, r]
    X = tf.concat(2, [tf.squeeze(x, [1]) for x in X])  # bsize, b, a*r, r
    X = tf.split(1, b, X)  # b, [bsize, a*r, r]
    X = tf.concat(2, [tf.squeeze(x, [1]) for x in X])  # bsize, a*r, b*r
    return tf.reshape(X, (bsize, a*r, b*r, 1))

poolio