From 4404793f81253ac9b1c971d98c417ef8dc8357d8 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Sun, 22 Jul 2018 20:38:57 +0200 Subject: [PATCH] define indices_inverse when indices is not None prob. doesn't matter but if `indices` is not `None` then `indices_inverse` is not set, so I moved it out of that `if` --- tfops.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tfops.py b/tfops.py index d978419..042a335 100644 --- a/tfops.py +++ b/tfops.py @@ -388,10 +388,11 @@ def shuffle_features(name, h, indices=None, return_indices=False, reverse=False) n_channels = int(h.get_shape()[-1]) indices = list(range(n_channels)) rng.shuffle(indices) - # Reverse it - indices_inverse = [0]*n_channels - for i in range(n_channels): - indices_inverse[indices[i]] = i + + # Reverse it + indices_inverse = [0]*n_channels + for i in range(n_channels): + indices_inverse[indices[i]] = i tf_indices = tf.get_variable("indices", dtype=tf.int32, initializer=np.asarray( indices, dtype='int32'), trainable=False)