PyTorch - 逐元素签名的最小值/最大值？

Question

我可能遗漏了一些明显的东西，但我找不到计算方法。

给定两个张量，我想保留每个张量中的最小元素以及符号。

我想过

sign_x = torch.sign(x)
sign_y = torch.sign(y)
min = torch.min(torch.abs(x), torch.abs(y))

为了最终将符号与获得的最小值相乘，但是我没有方法将正确的符号与保留的每个元素相乘，并且必须选择两个张量之一。

Answer 1

这是一种方法。将torch.sign(x) 和torch.sign(y) 乘以表示x 或y 是否是min 计算结果的布尔张量。然后将两个结果张量的逻辑或 (|) 组合起来，然后乘以 min 计算。

mins = torch.min(torch.abs(x), torch.abs(y))

xSigns = (mins == torch.abs(x)) * torch.sign(x)
ySigns = (mins == torch.abs(y)) * torch.sign(y)
finalSigns = xSigns.int() | ySigns.int()

result = mins * finalSigns

如果x和y对于某个元素具有相同的绝对值，则在上面的代码中x的符号优先。要让y 优先，请交换顺序并改用finalSigns = ySigns.int() | xSigns.int()。