sk1712/gcn_metric_learning

tf.concat arguments axis first and values second

xuesongwang opened this issue · 0 comments

Hi there,
First of all, thank you very much for this SUPER amazing project, you have no idea how much efforts I have spent on data preprocessing for Brain Network datasets before I saw this repo,
Just a minor issue regarding tf.concat function here, for my tensorflow version (1.15.0 + ) the order of arguments should be:
tf.concat( values, axis, name='concat' )
and there are a few places which I refer down below could cause errors, but it shouldn't be hard to find:

return tf.concat(0, [x, x_]) # K x M x Fin*N

return tf.concat(0, [x, x_]) # K x M x Fin*N

x = tf.concat(1, [x, site])

Hope that's helpful :)