Solved by SimonW in post #2 
              
              
                PyTorch uses dynamic graph. So you can disregard all those confusing symbolic variables from tf.  Just put whatever size you get from that batch of data.