如何在Pytorch中使用Tensorboard只需画一个简单的网

Question

我是一位新的徒。 我想画一个非常简单的网，只是一个输入，它具有恒定的权重和偏差。 这个网只是a = w * x + b。每个输入都是标量。 我想使用pytorch和tensorboard来显示此网络。

我在下面写代码。

class Net(nn.Module):
    def __init__(self, w, b):
        super(Net, self).__init__()
        self.w = w
        self.b = b

    def forward(self, x):
        x = self.w * x + self.b
        return x


net = Net(1.3, 3.0)

writer = SummaryWriter('runs/CH1_2_1')
input = torch.tensor([1.0], dtype=torch.float)
input.requires_grad_(True)
writer.add_graph(net, input)
writer.close()

我想要的网不应该是这样的 有什么问题