Tensorflow feature map