Function dot

p2m/layers.py:71–77

Wrapper for tf.matmul (sparse vs dense).

(x, y, sparse=False)

def dot(x, y, sparse=False):
    """Wrapper for tf.matmul (sparse vs dense)."""
    if sparse:
        res = tf.sparse_tensor_dense_matmul(x, y)
    else:
        res = tf.matmul(x, y)
    return res


class Layer(object):