Функция PyTorch torch.div
Справочное руководство Pytorch torch
torch.divЭто функция в PyTorch, используемая для поэлементного деления.
Определение функции
torch.div(input, other, rounding_mode, out)
Пример использования
Пример
import torch
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# Поэлементное деление
c = torch.div(a, b)
print(c)
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# Поэлементное деление
c = torch.div(a, b)
print(c)
Результат вывода:
tensor([5., 5., 6.])
Другие расширения