tree c2a1210c12b4682ab60bb47136580f249975a8b8
parent ae76544efc2731acb3f1ab401ffa1a35c02f57d6
author Yong Tang <yong.tang.github@outlook.com> 1588606739 +0000
committer Yong Tang <yong.tang.github@outlook.com> 1588606739 +0000

Fix issue in boolean_mask when axis is passed as a tensor

This PR tries to address the issue raised in 32236 where
a TypeError was thrown out when axis is passed as a tensor.
In the docstring axis has been specified as accepting a 1-D tensor.

This PR fixes the issue.

This PR fixes 32236.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>