当我在pytorch中使用tensorboardX时,我对GRAPHS中的'input'感到困惑,这是我的代码:
writer = SummaryWriter(log_dir='runs/testRes18/Restest18')
net = nn.Linear(512,10,bias=False)
input_tensor = torch.Tensor(10, 512)
writer.add_graph(net, input_tensor)
在我的GRAPHS中,预期一个输入(10 * 512)和一个分支位于Linear中,但是我在输入中找到了另一个常数5,在Linear中找到了两个分支。 有人可以解释吗?如何理解“输入”的含义?等于网络的输入吗?
↓Screecut
enter image description here