torch.hypot
-
torch.hypot(input, other, *, out=None) → Tensor
给定直角三角形的两条直角边,返回其斜边。
outi=inputi2+otheri2 input
和 other
的形状必须可广播。
- 参数:
输入(张量)- 第一个输入张量
其他(张量)- 第二个输入张量
- 关键字参数:
输出(张量,可选)- 输出张量。
示例:
>>> a = torch.hypot(torch.tensor([4.0]), torch.tensor([3.0, 4.0, 5.0]))
tensor([5.0000, 5.6569, 6.4031])