Return a tweaked version of 'axis'.
(parameters)
| 4474 | }] |
| 4475 | |
| 4476 | def get_valid_axis(parameters): |
| 4477 | """Return a tweaked version of 'axis'.""" |
| 4478 | axis = parameters["axis"] |
| 4479 | shape = parameters["base_shape"][:] |
| 4480 | while axis > len(shape) - 1: |
| 4481 | axis -= 1 |
| 4482 | return axis |
| 4483 | |
| 4484 | def build_graph(parameters): |
| 4485 | input_tensor = tf.placeholder( |