Understanding Tensorflow using Go