| 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)) |
#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