Return a rank if it is a tensor, else return None.
(tensor)
| 34 | |
| 35 | |
| 36 | def rank(tensor): |
| 37 | """Return a rank if it is a tensor, else return None.""" |
| 38 | if isinstance(tensor, ops.Tensor): |
| 39 | return tensor._rank() # pylint: disable=protected-access |
| 40 | return None |
| 41 | |
| 42 | |
| 43 | def scalar_shape(unused_op): |