x：待分割张量
axis：分割的维度索引号
num_or_size_splits：切割方案
当num_or_size_splits为单个数值时，如10，表示切割 为 10 份
当 num_or_size_splits...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...()
# 方式1
tf.boolean_mask(x,mask) # 通过掩码的方式获取数据
# 方式2
indices=tf.where(mask) # 提取所有大于 0 的元素索引
tf.gather_nd
云服务器
ICP备案
对象存储
腾讯会议
云直播
