如何理解tensorboardX-GRAPHS中'input'的含义?

时间:2019-08-27 07:41:46

标签: pytorch tensorboard tensorboardx

当我在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

0 个答案:

没有答案