Question

我之前已经问过这个问题Create boolean mask on TensorFlow，关于如何仅在某些索引设置为1时获取张量，其余的设置为0。

我认为@MZHm给出的答案将完全解决我的问题。虽然dense_shape的参数tf.SparseTensor只接受列表，但我想传递一个从图形中推断出来的形状（来自另一个具有可变形状的张量的形状）。所以在我的具体情况中，我想做这样的事情：

# The tensor from which the shape of the sparse tensor is to be inferred
reference_t = tf.zeros([32, 50, 11])

# The indices that will be 1
indices = [[0, 0],
           [3, 0],
           [5, 0],
           [6, 0]]

# Just setting all the values for the sparse tensor to be 1
values = tf.ones([reference_t.shape[-1]])

# The 2d shape I want the sparse tensor to have
sparse_2d_shape = [reference_t.shape[-2],
                   reference_t.shape[-1]]

st = tf.SparseTensor(indices, values, sparse_2d_shape)

由此我得到错误：

TypeError：预期为int64，得到'Dimension'类型的Dimension（50）代替。

如何动态设置稀疏张量的形状？是否有更好的选择来实现我的目标？

Answer 1

以下是您可以采取的动态形状：

import tensorflow as tf 
import numpy as np

indices = tf.constant([[0, 0],[1, 1]], dtype=tf.int64)
values = tf.constant([1, 1])
dynamic_input = tf.placeholder(tf.float32, shape=[None, None])
s = tf.shape(dynamic_input, out_type=tf.int64)

st = tf.SparseTensor(indices, values, s)
st_ordered = tf.sparse_reorder(st)
result = tf.sparse_tensor_to_dense(st_ordered)

sess = tf.Session()

具有（动态）形状[5, 3]的输入：

sess.run(result, feed_dict={dynamic_input: np.zeros([5, 3])})

将输出：

array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 0],
       [0, 0, 0],
       [0, 0, 0]], dtype=int32)

具有（动态）形状[3, 3]的输入：

sess.run(result, feed_dict={dynamic_input: np.zeros([3, 3])})

将输出：

array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 0]], dtype=int32)

所以你去...动态稀疏的形状。

TensorFlow SparseTensor，动态设置dense_shape

1 个答案: