如果希望在某个维度上全部按照长度为1进行分割,使用tf.unstack,切割长度固定为1。...tf.where
通过tf.where(cond, a, b)操作可以根据cond条件的真假从a 或 b 中读取数据
当a=b=None即 a,b 参数不指定时,``tf.where会返回cond张量中所有...,False],[False,True,False],[True,True,False]])
tf.where(cond,a,b) # 根据条件从 a,b 中采样
tf.where(cond) # 获取...cond 中为 True 的元素索引
demo
获取张量中的正数及其索引
x = tf.random.normal([3,3]) # 构造 a
mask=x>0 # 比较操作,等同于 tf.equal...(x,indices) # 拿到索引后,提取正数的元素值
scatter_nd
通过tf.scatter_nd(indices, updates,shape)实现刷新张量的部分数据。