Viewing a single comment thread. View all comments

saw79 t1_izjv9wa wrote

Sometimes that beautiful one-liner just isn't worth it compared to something like

torch.cat((
    img[:N, :N],
    img[N:, :N],
    img[:N, N:],
    img[N:, N:],
    ), dim=-1)
3