分析torch.norm

torch.norm的本质公式是:
f ( x , p ) = ∣ ∣ x ∣ ∣ p = ∑ i = 1 N x i p p f(x, p)=||x||_p= \sqrt[p]{\sum_{i=1}^Nx_i^p} f(x,p)=∣∣xp=pi=1Nxip

默认情况下, p p p的值取2

我们在torch.norm源码中找到一个例子

>>> import torch
>>> a = torch.arange(9, dtype= torch.float) - 4
>>> torch.norm(a)
tensor(7.7460)

来验证一下我们的本质公式。

a = [i-4 for i in range(9)]
s = 0
p = 2
for i in a:
    s += i**p
f_x = pow(s, 1/p)
print(f_x)

最后求得结果为:

7.745966692414834

约等于7.7460

你可能感兴趣的:(python,开发语言,人工智能,算法)