Функция PyTorch torch.divide

Функция PyTorch torch.divide


Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.divideЭто функция в PyTorch для поэлементного деления (эквивалентна div).

Определение функции

torch.divide(input, other, *, rounding_mode=None, out=None)

Пример использования

Пример

import torch

# Пример поэлементного деления
x = torch.tensor([10.0, 20.0, 30.0])
y = torch.tensor([2.0, 4.0, 5.0])
result = torch.divide(x, y)
print(result)  # tensor([5.0000, 5.0000, 6.0000])

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения