Pytorch如何对3D Tensor（变量）中的每个矩阵进行行归一化？

Question

如果我有一个尺寸为[a，b，c]的3D张量（变量）。将其视为ab * c矩阵，我希望所有这些矩阵都进行行归一化。

Answer 1

您可以使用该normalize功能。

import torch.nn.functional as f
f.normalize(input, p=2, dim=2)

该dim=2参数告诉沿着该维度正常化（除以其每一行向量p-norm。