nn.Linear输出五个坐标值和边框大小的代码

以下是一个简单的 PyTorch 代码示例，使用 nn.Linear 输出五个坐标值和边框大小：

import torch
import torch.nn as nn

# 创建一个输入张量，形状为 (batch_size, input_size)
batch_size = 32
input_size = 64
input_tensor = torch.randn(batch_size, input_size)

# 创建一个线性层，输入大小为 input_size，输出大小为 5
linear_layer = nn.Linear(input_size, 5)

# 将输入张量输入线性层，得到输出张量，形状为 (batch_size, 5)
output_tensor = linear_layer(input_tensor)

# 打印输出张量的形状
print(output_tensor.shape)

这个代码示例中，我们首先创建了一个形状为 (batch_size, input_size) 的输入张量 input_tensor，其中 batch_size 表示输入的批次大小，input_size 表示输入的特征维度。然后我们创建了一个线性层 linear_layer，它的输入大小为 input_size，输出大小为 5。将输入张量 input_tensor 输入到线性层中，得到输出张量 output_tensor，它的形状为 (batch_size, 5)。最后我们打印了输出张量的形状。