x.norm(p=2,dim=1,keepdim=True)

news/2025/2/12 15:20:48/

代码:x.norm(p=2,dim=1,keepdim=True)

功能:求指定维度上的范数。

函数原型:【返回输入张量给定维dim 上每行的p范数】

torch.norm(input, p, dim, out=None,keepdim=False) → Tensor

注:范数求法:【对N个数据求p范数】
||x||p=p次根号下 (x1的p次幂 + x2的p次幂 + x3的p次幂 + …)

input (Tensor) – 输入张量

p (float) – 范数计算中的幂指数值

dim (int) – 缩减的维度,dim=0是对0维度上的一个向量求范数,返回结果数量等于其列的个数,也就是说有多少个0维度的向量, 将得到多少个范数。dim=1同理。

out (Tensor, optional) – 结果张量

keepdim(bool)– 保持输出的维度 。当keepdim=False时,输出比输入少一个维度(就是指定的dim求范数的维度)。
而keepdim=True时,输出与输入维度相同,仅仅是输出在求范数的维度上元素个数变为1。这也是为什么有时
我们把参数中的dim称为缩减的维度,因为norm运算之后,此维度或者消失或者元素个数变为1。

import torchinputs=torch.tensor([[ 1.,  2.,  3.,  4.],[ 2.,  4.,  6.,  8.],[ 3.,  6.,  9., 12.]])
print(inputs.shape)#torch.Size([3, 4])
inputs1 = torch.norm(inputs, p=2, dim=1, keepdim=True)
print(inputs1)

torch.Size([3, 4])==》torch.Size([3, 1])
tensor([[ 5.4772],
[10.9545],
[16.4317]])
求解过程:dim=1即就是在4个数的维度上进行求范数
5.4772 = 2√ ̄(12+22+32+42)
10.9545 = 2√ ̄(22+42+62+82)
16.4317=2√ ̄(32+62+92+122)

inputs2 = torch.norm(inputs, p=2, dim=0, keepdim=True)
print(inputs2)

torch.Size([3, 4])==》torch.Size([1, 4])
tensor([[ 3.7417, 7.4833, 11.2250, 14.9666]])
计算过程:dim=0,即就是在3行的维度上进行纵向求解,在哪个维度上,输出结构哪个维度将会变成1
3.7417 = 2√ ̄(12+22+3^2)
7.4833 = 2√ ̄(22+42+6^2)
11.2250 = 2√ ̄(32+62+9^2)
14.9666 = 2√ ̄(42+82+12^2)


http://www.ppmy.cn/news/142575.html

相关文章

2250. 统计包含每个点的矩形数目

题目描述: 给你一个二维整数数组 rectangles ,其中 rectangles[i] [li, hi] 表示第 i 个矩形长为 li 高为 hi 。给你一个二维整数数组 points ,其中 points[j] [xj, yj] 是坐标为 (xj, yj) 的一个点。 第 i 个矩形的 左下角 在 (0, 0) 处&…

POJ 2250 Compromise (UVA 531)

LCS问题。基金会DP。 我很伤心WA非常多。就在LCS问题,需要记录什么路。 反正自己的纪录path错误,最后,就容易上当。 没有优化,二维阵列,递归打印,cin.eof() 来识别 end of file 标识。 至于单词用map 映射的…

poj 2250 Compromise(区间dp)

题目链接&#xff1a;http://poj.org/problem?id2250 思路分析&#xff1a;最长公共子序列问题的变形&#xff0c;只是把字符变成了字符串&#xff0c;按照最长公共子序列的思路即可以求解。 代码如下&#xff1a; #include <stdio.h> #include <string.h>#defin…

Poj-2250-Compromise

题意是找两篇文章中的最长子单词序列 能得出个数&#xff0c;但不知如何输出&#xff0c;找不到路径 看了别人的dfs&#xff0c;有所领悟&#xff1a; 若输入s1&#xff1a;ab,bd,fk,ce,ak,bt,cv s2: ab,fk,ce,tt,ak,bt,深搜路径数字涂红dp棋盘如下&#xff1a;    ab…

洛谷—— P1238 走迷宫

https://www.luogu.org/problem/show?pid1238 题目描述 有一个m*n格的迷宫(表示有m行、n列)&#xff0c;其中有可走的也有不可走的&#xff0c;如果用1表示可以走&#xff0c;0表示不可以走&#xff0c;文件读入这m*n个数据和起始点、结束点(起始点和结束点都是用两个数据来描…

题目 2250: 蓝桥杯算法提高-秘密行动

题目 小D接到一项任务&#xff0c;要求他爬到一座n层大厦的顶端与神秘人物会面。这座大厦有一个神奇的特点&#xff0c;每层的高度都不一样&#xff0c;同时&#xff0c;小D也拥有一项特殊能力&#xff0c;可以一次向上跳跃一层或两层&#xff0c;但是这项能力无法连续使用。已…

【DP】poj2250

经典的lcs问题&#xff0c;然而却是得到了很多启发&#xff0c;当时并没有想到如果s[i]ss[j]就可以直接用dp[i-1][j-1]1转移&#xff0c;仔细想了一下&#xff0c;dp[i][j-1]和dp[i-1][j]一定是<dp[i-1][j-1]1的。如果i在之前的状态中被使用过&#xff0c;那么dp[i][j-1]dp[…

【LOJ2250】「ZJOI2017」仙人掌

【题目链接】 点击打开链接 【思路要点】 考虑给定的图为大小为 N 1 N1 N1 的点的菊花图的情况&#xff0c;记方案数为 d p N dp_N dpN​ &#xff0c;考虑最后一个儿子是否连边&#xff0c;则有转移 d p i d p i − 1 ( i − 1 ) d p i − 2 dp_idp_{i-1}(i-1)dp_{i-2} …