帮我生成一段 pytorch 的简单代码

当然可以，以下是一个简单的 PyTorch 代码示例：

import torch

# Create a random tensor of shape 3x3
x = torch.randn(3, 3)

# Create a linear layer with 3 input features and 1 output feature
linear_layer = torch.nn.Linear(3, 1)

# Feed the tensor through the linear layer
output = linear_layer(x)

print(output)

这段代码创建了一个随机的 3x3 张量并创建了一个包含一个线性层的神经网络。张量被输入到这个网络中，最后输出一个形状为 (3, 1) 的张量。您可以将此示例代码作为起点，以构建您的 PyTorch 模型。