A neural network approach for solving the Monge-Amp\`ere equation with transport boundary condition