Loading deepchem/nn/layers.py +8 −9 Original line number Diff line number Diff line Loading @@ -1142,8 +1142,7 @@ class DAGLayer(Layer): batch_outputs = self.DAGgraph_step(batch_inputs, self.W_list, self.b_list) # index for targe atoms target_index = tf.stack( [tf.range(n_atoms), parents[:, count, 0]], axis=1) target_index = tf.stack([tf.range(n_atoms), parents[:, count, 0]], axis=1) # index for dummies target_index2 = tf.stack( [tf.range(n_atoms), tf.constant(self.max_atoms, shape=(n_atoms,))], Loading Loading
deepchem/nn/layers.py +8 −9 Original line number Diff line number Diff line Loading @@ -1142,8 +1142,7 @@ class DAGLayer(Layer): batch_outputs = self.DAGgraph_step(batch_inputs, self.W_list, self.b_list) # index for targe atoms target_index = tf.stack( [tf.range(n_atoms), parents[:, count, 0]], axis=1) target_index = tf.stack([tf.range(n_atoms), parents[:, count, 0]], axis=1) # index for dummies target_index2 = tf.stack( [tf.range(n_atoms), tf.constant(self.max_atoms, shape=(n_atoms,))], Loading
