def build_graph(self):
# Build graph
state = graph.Placeholder(np.float32, shape=(2, ))
reverse = graph.TfNode(tf.reverse(state.node, [0]))
# Expose public API
self.op_get_action = self.Op(reverse, state=state)