Splits first dimension into [first_dim, -1].
(x, first_dim)
| 1467 | |
| 1468 | |
| 1469 | def _unflatten_first_dim(x, first_dim): |
| 1470 | """Splits first dimension into [first_dim, -1].""" |
| 1471 | old_shape = array_ops.shape(x) |
| 1472 | new_shape = array_ops.concat([first_dim, [-1], old_shape[1:]], axis=0) |
| 1473 | return array_ops.reshape(x, new_shape) |
| 1474 | |
| 1475 | |
| 1476 | def _inputs_with_flattening(pfor_input, input_indices): |
no test coverage detected