(output_)
| 4062 | |
| 4063 | # static shape inference |
| 4064 | def set_shape(output_): |
| 4065 | if isinstance(output_, ops.Tensor): |
| 4066 | shape = output_.shape.as_list() |
| 4067 | shape[0] = time_steps |
| 4068 | shape[1] = batch |
| 4069 | output_.set_shape(shape) |
| 4070 | return output_ |
| 4071 | |
| 4072 | outputs = nest.map_structure(set_shape, outputs) |
| 4073 |