假设我有N个元素，并且知道如何连接。这意味着有一个向量[c1，c2，...，cn]，其中[0 ... CC] CC-分量中的ci计数。
我想从中获得邻接矩阵。
简单示例输入
[0, 0, 1, 2, 0, 1]
输出
1 1 0 0 1 0
1 1 0 0 1 0
0 0 1 0 0 1
0 0 0 1 0 0
1 1 0 0 1 0
0 0 1 0 0 1
是否可以仅使用线性代数加上TF中的一些额外的张量运算来获得它？
可以简单地通过与自身的广播比较来完成：
import tensorflow as tf
input = [0, 0, 1, 2, 0, 1]
adjacency_matrix = tf.cast(tf.equal(tf.expand_dims(input, 1), input), tf.int32)
print(sess.run(adjacency_matrix))
输出：
[[1 1 0 0 1 0]
[1 1 0 0 1 0]
[0 0 1 0 0 1]
[0 0 0 1 0 0]
[1 1 0 0 1 0]
[0 0 1 0 0 1]]
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句