Method decode

Args: epsilon: Tensor epslion [batch, tgt_length, nz] tgt_mask: Tensor tensor of target masks [batch, tgt_length] src: Tensor source encoding [batch, src_length, hidden_size] src_mask: Tensor

(self, epsilon: torch.Tensor, tgt_mask: torch.Tensor,
               src: torch.Tensor, src_mask: torch.Tensor)