torch.tril_indices

torch.tril_indices(row, col, offset=0, *, dtype=torch.long, device='cpu', layout=torch.strided) → Tensor

Возвращает индексы нижней треугольной части матрицы размером row на col в тензоре размером 2 на N, где первая строка содержит координаты строк всех индексов, а вторая строка — координаты столбцов. Индексы упорядочены сначала по строкам, затем по столбцам.

Нижняя треугольная часть матрицы определяется как элементы на главной диагонали и ниже неё.

Аргумент offset определяет, какие диагонали учитывать. Если offset = 0, сохраняются все элементы на главной диагонали и ниже неё. Положительное значение включает столько же диагоналей над главной диагональю, а отрицательное значение исключает столько же диагоналей под главной диагональю. Главная диагональ — это набор индексов $\lbrace (i, i) \rbrace$ для $i \in [0, \min\{d_{1}, d_{2}\} - 1]$, где $d_{1}, d_{2}$ — размеры матрицы.

Примечание

При работе с CUDA значение row * col должно быть меньше $2^{59}$, чтобы избежать переполнения при вычислениях.

Параметры:

• row (int) – количество строк в двумерной матрице.
• col (int) – количество столбцов в двумерной матрице.
• offset (int) – смещение диагонали относительно главной диагонали. По умолчанию: если не указано, 0.

Именованные аргументы:

• dtype (torch.dtype, необязательно) – требуемый тип данных возвращаемого тензора; поддерживаются только torch.int, torch.long. По умолчанию: если None, torch.long.
• device (torch.device, необязательно) – требуемое устройство для возвращаемого тензора. По умолчанию: если None, используется текущее устройство для типа тензора по умолчанию (см. torch.set_default_device()). device будет CPU для типов тензоров CPU и текущим устройством CUDA для типов тензоров CUDA.
• layout (torch.layout, необязательно) – в настоящее время поддерживается только torch.strided.

Пример:

>>> a = torch.tril_indices(3, 3)
>>> a
tensor([[0, 1, 1, 2, 2, 2],
        [0, 0, 1, 0, 1, 2]])

>>> a = torch.tril_indices(4, 3, -1)
>>> a
tensor([[1, 2, 2, 3, 3, 3],
        [0, 0, 1, 0, 1, 2]])

>>> a = torch.tril_indices(4, 3, 1)
>>> a
tensor([[0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3],
        [0, 1, 0, 1, 2, 0, 1, 2, 0, 1, 2]])