def past_shape()

in src/model.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

def past_shape(*, hparams, batch_size=None, sequence=None):
    return [batch_size, hparams.n_layer, 2, hparams.n_head, sequence, hparams.n_embd // hparams.n_head]