想要求pytorch tensor中某个2048维度的向量的模长，可以先相乘，然后再用sum求和。
假设 v是一个2048维的向量，则可以利用一下两个语句求出模长的平方。
sq = v * v
sum_sq = sq.sum()
pytorch tensor求向量的模长
猜你喜欢
转载自blog.csdn.net/t20134297/article/details/103817568
今日推荐
周排行