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.])
その他の拡張