擅长:python、mysql、java
<p>让<code>sp</code>作为2d SparseTensor的名称。首先可以为要提取的SparseTensor行创建一个指示符张量,即</p>
<pre><code>mask = tf.concat([tf.constant([True, True]), tf.fill([sp.dense_shape[0] - 2],
False)], axis=0)
</code></pre>
<p>下次使用tf.聚集要将其传播到稀疏索引,请执行以下操作:</p>
^{pr2}$
<p>最后</p>
<pre><code>values = tf.boolean_mask(sp.values, mask_sp)
indices = tf.boolean_mask(sp.indices, mask_sp)
dense_shape = [sp.dense_shape[0] - 2, sp.dense_shape[1]]
output_sp = tf.SparseTensor(indices=indices, values=values, dense_shape=dense_shape)
</code></pre>