在“ of torch.autograd.grad”中指出,对于参数,
输出(张量序列)-差异化函数的输出。输入(张量序列)-输入w.r.t.其中的梯度将被返回(而不是累积到.grad中)。torch.tensor([a, b])) #RuntimeError: grad can be implicitly created only for scalar outputs
我想得到张量列表的梯度输入<
我试图用两个输出建立一个模型,一个用于分类,另一个用于回归。我使用来自Keras的预先训练过的VGG16作为卷积特征提取器。但是,当我试图在两个输出序列中构建下一个平坦层时,我得到的错误是输入不是张量。如何确保我只将卷积基的最后一层作为下一层的输入?tf.AUTO_REUSE):
x = Flatten()(conv_base.layer