I need to convert following tensorflow model into pytorch model. Any help?