(x)
| 385 | input_flatten = lambda x: nest.flatten(x) if input_is_sequence else [x] |
| 386 | |
| 387 | def input_pack(x): |
| 388 | return nest.pack_sequence_as(elems, x) if input_is_sequence else x[0] |
| 389 | |
| 390 | if initializer is None: |
| 391 | output_is_sequence = input_is_sequence |