替换tensorflow python中的值或创建张量的掩码

tf.where(condition, x, y)

import tensorflow as tf

# 创建一个张量
tensor = tf.constant([1, 2, 3, 4, 5])

# 替换值
replaced_tensor = tf.where(tensor > 3, 10, tensor)
print(replaced_tensor.numpy())  # 输出：[1 2 3 10 10]

# 创建掩码
mask = tf.where(tensor > 3, True, False)
print(mask.numpy())  # 输出：[False False False  True  True]

tf.boolean_mask(tensor, mask)

import tensorflow as tf

# 创建一个张量
tensor = tf.constant([1, 2, 3, 4, 5])

# 创建掩码
mask = tensor > 3

# 使用掩码创建新的张量
masked_tensor = tf.boolean_mask(tensor, mask)
print(masked_tensor.numpy())  # 输出：[4 5]