weixin_44912950的博客

读懂神经网络代码

在PyTorch中，张量（Tensor）是多维数组，其中包含了多个元素。维度用于描述数据在各个方向上的大小或者说是用来描述张量中数据排列的方式。可以将张量想象成一个多维网格，其中每个网格上的元素都对应着一个数值。一个张量的维度数量被称为它的“阶”或“秩”，而每个维度都有一个大小（size），表示该维度上可以容纳的元素数量。在通道注意力模块代码中，维度是指张量在每个轴上的大小。属性来获取张量的形状，这将返回一个包含各维度大小的元组。维度上与其他张量保持一致，以便后续的计算可以正确进行。，表示有3行和4列。

pytorch学习知识