问题来自此文件的第138行 https://github.com/tkipf/gcn/blob/master/gcn/models.py
假设模型的__init__
函数具有输入占位符(二维Matirx),我希望通过以下代码获得张量的尺寸
self.input_dim = self.inputs.shape[1].value
或self.input_dim = self.inputs.get_shape().as_list()[1]
__init__
函数起作用,并且通过调试代码,我发现它们立即返回int
类型的正确尺寸。
但是,在训练过程中,发生以下错误,
ValueError:Tensor Tensor(“ Const:0”,shape =(2,),dtype = int64)可能无法馈送。
如果您手动设置self.input_dim
,则该培训有效。
令我更加困惑的是,第139行(紧接有问题的第138行之后)有效:
self.output_dim = placeholders['labels'].get_shape().as_list()[1]
似乎只是对另一个张量做同样的事情,但这行有效!
希望有人可以帮助解释这里发生的事情。您可以只下载整个存储库并运行train.py
文件。通过注释或取消注释model.py
的第138行,可以重现此问题。