TensorFlow-ESPCN icon indicating copy to clipboard operation
TensorFlow-ESPCN copied to clipboard

Problem with the _phase_shift() function

Open hx450652229 opened this issue 5 years ago • 0 comments

I suppose there's a little bug in the _phase_shift function, cause if bsize is 1 then after tf.squeeze(x) the batch dimension would also disappear.

Maybe tf.squeeze(x, 1) instead?

hx450652229 avatar May 06 '19 10:05 hx450652229