stomach_pytorch