Computes the number of elements in `losses` tensor.
(losses)
| 50 | |
| 51 | |
| 52 | def _num_elements(losses): |
| 53 | """Computes the number of elements in `losses` tensor.""" |
| 54 | with K.name_scope('num_elements') as scope: |
| 55 | return math_ops.cast(array_ops.size(losses, name=scope), dtype=losses.dtype) |
| 56 | |
| 57 | |
| 58 | def reduce_weighted_loss(weighted_losses, |
no test coverage detected