Question

我有一些稀疏索引：

<key>FacebookAppID</key>
<string>FACEBOOK_APP_ID</string>
<key>FacebookDisplayName</key>
<string>FACEBOOK_APP_NAME</string>

每个索引的对应值为：

[[0 0]
 [0 1]
 [1 0]
 [1 1]
 [1 2]
 [2 0]]

如何在张量流中将6x3值张量转换为3x3x3密集张量？未在索引中指定的索引值为零向量[0。 0. 0.]。密集的张量就是这样：

[[0.1 0.2 0.3]
 [0.4 0.5 0.6]
 [0.7 0.8 0.9]
 [1.0 1.1 1.2]
 [1.3 1.4 1.5]
 [1.6 1.7 1.8]]

Answer 1

您可以使用tf.scatter_nd来做到这一点：

import tensorflow as tf

with tf.Graph().as_default(), tf.Session() as sess:
    indices = tf.constant(
        [[0, 0],
         [0, 1],
         [1, 0],
         [1, 1],
         [1, 2],
         [2, 0]])
    values = tf.constant(
        [[0.1, 0.2, 0.3],
         [0.4, 0.5, 0.6],
         [0.7, 0.8, 0.9],
         [1.0, 1.1, 1.2],
         [1.3, 1.4, 1.5],
         [1.6, 1.7, 1.8]])
    out = tf.scatter_nd(indices, values, [3, 3, 3])
    print(sess.run(out))

输出：

[[[0.1 0.2 0.3]
  [0.4 0.5 0.6]
  [0.  0.  0. ]]

 [[0.7 0.8 0.9]
  [1.  1.1 1.2]
  [1.3 1.4 1.5]]

 [[1.6 1.7 1.8]
  [0.  0.  0. ]
  [0.  0.  0. ]]]

Answer 2

在Tensorflow中没有使用任何reshape类型函数的明确方法。我只能通过创建列表并将其转换回张量来考虑迭代解决方案。这可能不是最有效的解决方案，但这可能适用于您的代码。

# list of indices 
idx=[[0,0],[0,1], [1,0],[1,1], [1,2], [2,0]]

# Original Tensor to reshape
dense_tensor=tf.Variable([[0.1, 0.2 ,0.3],[0.4, 0.5, 0.6], [0.7, 0.8, 0.9], [1.0,1.1,1.2],[1.3,1.4,1.5], [1.6,1.7,1.8]])

# creating a temporary list to later convert to Tensor
c=np.zeros([3,3,3]).tolist()

for i in range(3):
    count=0
    for j in range(3):
        if([i,j] in idx):
            c[i][j]=dense_tensor[count]
            count=count+1
        else:
            c[i][j]=tf.Variable([0,0,0], dtype=tf.float32)

# Convert obtained list to Tensor
converted_tensor = tf.convert_to_tensor(c, dtype=tf.float32)

您可以根据所需的张量的大小定义范围。对于您的情况，我选择了3个，因为您想要一个3x3x3张量。我希望这会有所帮助！

具有向量值到密集张量的Tensorflow稀疏张量

2 个答案: