Python PyTorch hypot用法及代码示例

本文简要介绍python语言中 torch.hypot 的用法。

用法:

torch.hypot(input, other, *, out=None) → Tensor

参数：

• input(Tensor) -第一个输入张量

• other(Tensor) -第二个输入张量

关键字参数：

out(Tensor,可选的) -输出张量。

给定一个直角三角形的边，返回它的斜边。

\text{out}_{i} = \sqrt{\text{input}_{i}^{2} + \text{other}_{i}^{2}}

input 和 other 的形状必须是可广播的。

例子：

>>> a = torch.hypot(torch.tensor([4.0]), torch.tensor([3.0, 4.0, 5.0]))
tensor([5.0000, 5.6569, 6.4031])