Function _sparse_eye

tf_euler/python/utils/sparse_aggregators.py:31–34

(num_rows, dtype=tf.float32)

def _sparse_eye(num_rows, dtype=tf.float32):
    return tf.SparseTensor(
        tf.stack([tf.range(num_rows)] * 2, axis=1),
        tf.ones(num_rows, dtype), tf.stack([num_rows] * 2))


class GCNAggregator(layers.Layer):