def __init__(self, rng, x, d): self.input = x self.out_len = d self.encoded_output = self.encode_final_state() ### default seq-to-seq model: tile C as input to all frames ###