tf.where()【能懂版】
用法1:tf.where(x, y, z)首先要求 x,y,z 的形状相同,返回的结果 res 也与它们形状相同,其中 x 是条件矩阵(数据类型为 tf.bool 的 tensor):若 x(i, j) 为 True,则 res(i, j) = y(i, j)若 x(i, j) 为 False,则 res(i, j) = z(i, j)用法2:tf.where(x)当 y 和 z 都为 None 时得到这种情况,其中 x 是条件矩阵(数据类型为 tf.bool 的 tensor),tf.wh
原创
2021-09-06 21:38:15 ·
944 阅读 ·
2 评论