Python tensorflow中稀疏到稠密的einsum

Python tensorflow中稀疏到稠密的einsum,python,performance,tensorflow,Python,Performance,Tensorflow,当我有复杂的操作时,有没有办法在稀疏矩阵上有效地执行操作?(事实上,我甚至不知道我的示例如何在没有切片的情况下简化为任何内容) 在本例中,我尝试使用一个复杂的内核向量对批次(b)中的多个输入进行分析,该向量对于批次中的每个示例都是不同的(因为它是由权重乘以另一个输入构建的)。输入是密集的,内核是稀疏的 例如: outputs = special_math_ops.einsum("bi,bnilo->bnlo", x, kernel) 因此,内核是一个tf.SparseTensor?@jd

当我有复杂的操作时,有没有办法在稀疏矩阵上有效地执行操作?(事实上,我甚至不知道我的示例如何在没有切片的情况下简化为任何内容)

在本例中,我尝试使用一个复杂的内核向量对批次(b)中的多个输入进行分析,该向量对于批次中的每个示例都是不同的(因为它是由权重乘以另一个输入构建的)。输入是密集的,内核是稀疏的

例如:

outputs = special_math_ops.einsum("bi,bnilo->bnlo", x, kernel)

因此,
内核
是一个
tf.SparseTensor
?@jdehesa yes还是一个0-1大部分为零的矩阵(如果这样表示更好的话)