Функция div

Функция div выполняет поэлементное деление тензоров. Первым параметром функция принимает делимое (тензор или скаляр), вторым параметром - делитель (тензор или скаляр). Функция поддерживает вещественные и целочисленные типы данных, а также автоматическое расширение формы (broadcasting).

Синтаксис

torch.div(dividend, divisor)

Пример

Давайте разделим два тензора одинаковой формы:

import torch

t1 = torch.tensor([10.0, 20.0, 30.0, 40.0])
t2 = torch.tensor([2.0, 4.0, 5.0, 8.0])
res = torch.div(t1, t2)
print(res)

Результат выполнения кода:

tensor([5.0000, 5.0000, 6.0000, 5.0000])

Пример

Давайте разделим тензор на скаляр:

import torch

t = torch.tensor([2.0, 4.0, 6.0, 8.0])
res = torch.div(t, 2.0)
print(res)

Результат выполнения кода:

tensor([1.0000, 2.0000, 3.0000, 4.0000])

Пример

Давайте разделим тензоры с целочисленными типами данных:

import torch

t1 = torch.tensor([10, 20, 30, 40])
t2 = torch.tensor([3, 6, 9, 12])
res = torch.div(t1, t2)
print(res)

Результат выполнения кода:

tensor([3, 3, 3, 3])

Пример

Давайте разделим двумерные тензоры с использованием broadcasting:

import torch

t1 = torch.tensor([
	[10.0, 20.0, 30.0],
	[40.0, 50.0, 60.0],
])
t2 = torch.tensor([2.0, 5.0, 10.0])
res = torch.div(t1, t2)
print(res)

Результат выполнения кода:

tensor([
	[5.0000, 4.0000, 3.0000],
	[20.0000, 10.0000, 6.0000],
])

Смотрите также

функцию mul,
которая выполняет поэлементное умножение тензоров
функцию add,
которая выполняет поэлементное сложение тензоров
функцию sub,
которая выполняет поэлементное вычитание тензоров
функцию true_divide,
которая выполняет деление с плавающей точкой