Solved by ptrblck in post #4 
              
              
                You don’t need to use placeholder variables in PyTorch and can directly pass tensors to the model without specifying their shapes beforehand (similar to what you would be using in numpy).