The network is theoretically implemented, but it produces a loss of NaN.

Tagged in order to debug the warning `WARNING:tensorflow:Using a while_loop for converting StridedSlice`