我有一个布尔张量，我想将其转换为由1和0组成的二进制张量。

为了说明情况，我有以下张量：

[[ True False  True] [False  True False] [ True False  True]]

我需要将其转换为1和0，以便我可以与一个值张量进行逐元素相乘，即：

[[1.         0.64082676 0.90568966] [0.64082676 1.         0.37999165] [0.90568966 0.37999165 1.        ]]

我尝试了以下两个函数：

   masks = tf.map_fn(logical, masks, dtype=tf.float32)   masks = tf.vectorized_map(logical, masks)

其中使用了：

@tf.functiondef logical(x):    if tf.equal(x, True):        return zero    return one

但遗憾的是没有成功。我也尝试过直接与布尔张量相乘，但这是不被允许的。

所以，请问如何解决这个问题？

回答：

我想我通过这个方法和一些“魔法”解决了这个问题。假设penalties是值张量

test = tf.where(masks, penalties * 0.0, penalties * 1.0)