张量的批量范数

Question

我有一个暗淡的 n x 3 张量 t。当我应用 torch.linalg.norm 时，它 return 是一个单一的值。我需要的是一个批量范数函数，它将 return 一个具有 n 范数的张量，一个对应 t 中的每个向量。

感谢您的帮助。

Answer 1

似乎最相关的文档位置是： https://pytorch.org/docs/stable/generated/torch.linalg.norm.html

您可以在终端中尝试：python3 然后执行以下 python 命令：

>>> from torch import linalg as LA
>>> c = torch.tensor([[1., 2., 3.],
...                   [-1, 1, 4]])
>>> LA.norm(c, dim=0)
tensor([1.4142, 2.2361, 5.0000])
>>> LA.norm(c, dim=1)
tensor([3.7417, 4.2426])

结论： 在您的具体情况下，您需要执行以下操作：

torch.linalg.norm(t,dim=1)

张量的批量范数

Batch-wise norm of a tensor

pytorch

tensor