pytorch tensor求向量的模长

想要求pytorch tensor中某个2048维度的向量的模长,可以先相乘,然后再用sum求和。

假设 v是一个2048维的向量,则可以利用一下两个语句求出模长的平方。

 sq = v * v
 sum_sq = sq.sum()

 

你可能感兴趣的:(pytorch)