def get_dims()

in iep/models/seq2seq.py [0:0]

• 11 lines of code
• 6 McCabe index (conditional complexity)

  def get_dims(self, x=None, y=None):
    V_in = self.encoder_embed.num_embeddings
    V_out = self.decoder_embed.num_embeddings
    D = self.encoder_embed.embedding_dim
    H = self.encoder_rnn.hidden_size
    L = self.encoder_rnn.num_layers

    N = x.size(0) if x is not None else None
    N = y.size(0) if N is None and y is not None else N
    T_in = x.size(1) if x is not None else None
    T_out = y.size(1) if y is not None else None
    return V_in, V_out, D, H, L, N, T_in, T_out