我目前正在尝试对 tensorflow 中的一对稀疏张量执行 kronecker 乘积。我在另一个线程上找到了一些代码来使用普通张量执行 kronecker 乘积。

def tf_kron(a,b):
    a_shape = [a.shape[0].value,a.shape[1].value]
    b_shape = [b.shape[0].value,b.shape[1].value]
    return tf.reshape(tf.reshape(a,[a_shape[0],1,a_shape[1],1])*tf.reshape(b,[1,b_shape[0],1,b_shape[1]]),[a_shape[0]*b_shape[0],a_shape[1]*b_shape[1]])

然而，稀疏张量并非如此。我不完全了解如何重塑稀疏张量。