AttributeError: Tensor object has no attribute _keras_history的解决方案

问题描述

在转写GIN (GRAPH ISOMORPHISM NETWORK)的pytorch代码为keras代码时，随手使用了K.relu( )函数，导致题目的报错，报错代码如下：

def next_layer(self, h, layer, adj_matrix=None):
     h2 = h
     pooled = Lambda(lambda x: K.batch_dot(x[0], x[1]))([adj_matrix, h2])
     pooled_rep = self.mlps[layer](pooled)
     h = self.batches[layer](pooled_rep)
 	 h = K.relu(h)
     return h

debug的过程中查看变量的值，发现变量h经过 K.relu( )之前存在_keras_history属性，经过K.relu( )后就没有了。

解决方法

1. 使用keras.backend内的函数
2. 使用keras.layers.Lambda

修改后代码：

def next_layer(self, h, layer, adj_matrix=None):
     h2 = h
     pooled = Lambda(lambda x: K.batch_dot(x[0], x[1]))([adj_matrix, h2])
     pooled_rep = self.mlps[layer](pooled)
     h = self.batches[layer](pooled_rep)
     h = Lambda(lambda x: K.relu(x))(h)
     return h

再次debug查看，成功。