refactor(autogram): Create block-diagonal matrix using einsum (#456)

refactor(autogram): Create block-diagonal matrix using einsum (#456) #149