Imported from https://github.com/TheAbhiKumar/tensorflow-value-iteration-networks