Функция PyTorch torch.copysign
Справочное руководство PyTorch torch
torch.copysign— это функция в PyTorch, используемая для поэлементного копирования знака.
Определение функции
torch.copysign(input, other, *, out=None)
Пример использования
Пример
import torch
# Пример копирования знака
x = torch.tensor([1.0, -2.0, 3.0, -4.0])
y = torch.tensor([-1.0, -1.0, 1.0, 1.0])
result = torch.copysign(x, y)
print(result) # tensor([-1., -2., 3., -4.])
# Пример копирования знака
x = torch.tensor([1.0, -2.0, 3.0, -4.0])
y = torch.tensor([-1.0, -1.0, 1.0, 1.0])
result = torch.copysign(x, y)
print(result) # tensor([-1., -2., 3., -4.])
Другие расширения