a, b = sess.run([c, d], feed_dict={…}), how to substitute this in pytorch?
In PyTorch you can directly use all operations without compiling a static graph and use sessions (similar to numpy, but with GPUs and Autograd).
Have a look at the tutorials for a quick start 