在TensorFlow中,您可以使用tf.random.uniform
函数生成随机布尔值
import tensorflow as tf
# 生成一个形状为(3, 3)的随机布尔张量
random_bool_tensor = tf.random.uniform(shape=(3, 3), minval=0, maxval=2, dtype=tf.int32)
# 将生成的整数张量转换为布尔张量
bool_tensor = tf.cast(random_bool_tensor, tf.bool)
print(bool_tensor)
此代码段首先生成一个形状为(3, 3)的随机整数张量,数值范围在0到1之间(包含0,不包含2)。接着使用tf.cast
函数将整数张量转换为布尔张量。生成的布尔张量将包含True
(对应于1)和False
(对应于0)。
领取专属 10元无门槛券
手把手带您无忧上云