title: tf.boolean_mask not accepting the axis argument
body: <p>Here is my code:  </p>
  
  <pre><code> 44     scores = tf.boolean_mask(box_class_scores,filtering_mask,axis=-1)
   45     boxes = tf.boolean_mask(boxes,filtering_mask,axis=-1)
   46     classes = tf.boolean_mask(box_classes,filtering_mask,axis=-1)
  </code></pre>
  
  <p>Error, I'm getting:</p>
  
  <blockquote>
    <p>TypeError: boolean_mask() got an unexpected keyword argument 'axis'</p>
  </blockquote>
  
  <p>The <code>tf.boolean_mask()</code> is not accepting the axis argument but is a valid argument as can be seen in the documentation: <a href="https://www.tensorflow.org/api_docs/python/tf/boolean_mask" rel="nofollow noreferrer">https://www.tensorflow.org/api_docs/python/tf/boolean_mask</a></p>
  
  <p><a href="https://i.stack.imgur.com/bne6B.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/bne6B.png" alt="enter image description here"></a></p>