nn.Linear的功能是？

nn.Linear是PyTorch中的一个类，用于定义线性变换的操作。它将输入张量与权重矩阵相乘，并加上偏置向量，然后输出结果。nn.Linear的功能可以总结为以下几点：

1. 定义线性变换：nn.Linear可以将输入张量与权重矩阵相乘，并加上偏置向量，从而实现线性变换操作。

2. 可变形状输入：nn.Linear的输入张量可以具有可变的形状，只需保证输入张量的维度满足[batch, *, in_features]的要求即可。

3. 速度较快：相比于nn.Conv1d中的卷积操作，当使用kernel_size=1时，nn.Linear的速度更快。

4. 等价计算：在进行等价计算时，需要保证nn.Linear的输入张量为三维，并且通过tensor.permute重新排列维度轴秩序，以实现与nn.Conv1d的等价计算。

下面是一个使用nn.Linear的示例代码：

import torch
import torch.nn as nn

# 定义输入张量
input_tensor = torch.randn(10, 20)  # 10个样本，个样本有20个特征

# 定义nn.Linear层
linear_layer = nn.Linear(20, 30)  # 输入特征数为20，输出特征数为30

# 进行线性变换
output_tensor = linear_layer(input_tensor)

# 输出结果
print(output_tensor.shape)  # 输出结果的形状为(10, 30)

在上述示例中，我们首先定义了一个输入张量input_tensor，它的形状为(10, 20)。然后，我们定义了一个nn.Linear层linear_layer，它的输入特征数为20，输出特征数为30。接下来，我们将输入张量通过linear_layer进行线性变换，得到输出张量output_tensor。最后，我们打印输出张量的形状，结果为(10, 30)。