Merges first two dimensions.
(x)
| 1460 | |
| 1461 | |
| 1462 | def _flatten_first_two_dims(x): |
| 1463 | """Merges first two dimensions.""" |
| 1464 | old_shape = array_ops.shape(x) |
| 1465 | new_shape = array_ops.concat([[-1], old_shape[2:]], axis=0) |
| 1466 | return array_ops.reshape(x, new_shape) |
| 1467 | |
| 1468 | |
| 1469 | def _unflatten_first_dim(x, first_dim): |
no test coverage detected