torch有必要做吗,torch项目组合

torch.flatten的作用是什么样的说明例子

解释

torch.flatten改变张量的维和维数，起到将指定维数之后的维数全部展开为一个维数的作用，新维数是所有展开维数的乘积。

首先，我们成为三维张量b，维数分别为3、1、3。 接下来，我们来看看从第0维展开和从第1维展开的结果：

如果您仔细比较它们，您可能会发现，您刚才提到的“将指定维之后的所有维的维数展开到一个维中，新维数是所有展开维的维数的乘积”。