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 ·
919 阅读 ·
2 评论