Skimming through the code it seems that tf.nn.moments can be replaced by torch.var, while other functions can be mapped to the torch. namespace without name changes.
Could you post your current approach and explain, where you are stuck at the moment?