Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"