import tensorflow as tf x=tf.constant([0.1, -1., 5.2, 4.3, -1., 7.4])#判斷x里面的元素是否是１ condition_mask=tf.not_equal(x,tf.constant(-1.))#[ True, False, True, True, False, True]#將張量拆成兩個(gè),按照condition_mask的對(duì)應(yīng)位置 partitioned_data = tf.dynamic_partition(x, tf.cast(condition_mask, tf.int32) , 2)#partitioned_data[0]=[-1., -1.] #partitioned_data[1]=[2.1, 7.2, 6.3, 9.4]partitioned_data[1] = partitioned_data[1] + 1.0#這行代碼是提取索引位置 condition_indices = tf.dynamic_partition(tf.range(tf.shape(x)[0]), tf.cast(condition_mask, tf.int32) , 2)x = tf.dynamic_stitch(condition_indices, partitioned_data) # Here x=[1.1, -1., 6.2, 5.3, -1, 8.4], the -1. values remain # unchanged.

總結(jié)

以上是生活随笔為你收集整理的tf.dynamic_stitch 和　 tf.dynamic_partition的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。