快捷方式

torch.renorm

torch.renorm(input, p, dim, maxnorm, *, out=None) Tensor

返回一个张量,其中 input 沿维度 dim 的每个子张量都进行了归一化,使得子张量的 p 范数小于值 maxnorm

注意

如果某行的范数小于 maxnorm,则该行保持不变

参数
  • input (Tensor) – 输入张量。

  • p (float) – 范数计算的幂

  • dim (int) – 用于切片以获取子张量的维度

  • maxnorm (float) – 保持每个子张量在下的最大范数

关键字参数

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

示例

>>> x = torch.ones(3, 3)
>>> x[1].fill_(2)
tensor([ 2.,  2.,  2.])
>>> x[2].fill_(3)
tensor([ 3.,  3.,  3.])
>>> x
tensor([[ 1.,  1.,  1.],
        [ 2.,  2.,  2.],
        [ 3.,  3.,  3.]])
>>> torch.renorm(x, 1, 0, 5)
tensor([[ 1.0000,  1.0000,  1.0000],
        [ 1.6667,  1.6667,  1.6667],
        [ 1.6667,  1.6667,  1.6667]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源