本文Permute层

Permute层是一种用于改变张量维度顺序的层。它可以将输入张量的维度重新排列，以满足特定的需求。在本文中，Permute层被用于将卷积层的输出张量的维度从(批次大小, 高度, 宽度, 通道数)转换为(批次大小, 通道数, 高度, 宽度)。

Permute层的作用是为了适应后续层的输入要求。在本文中，Permute层的输出被用作RNN层的输入。RNN层需要输入的维度顺序为(批次大小, 时间步长, 特征数)，而卷积层的输出维度顺序为(批次大小, 特征数, 高度, 宽度)。因此，通过使用Permute层，可以将卷积层的输出维度重新排列为RNN层所需的维度顺序。