Cadene

import time
import torch
import itertools

def make_pairs_ids(nregion, bsize):
    pairs_ids = []
    for batch_id in range(bsize):
        pairs_id = torch.LongTensor([
            (batch_id,i,j) for i,j in \
            itertools.product(range(nregion),repeat=2)])