tensorflow26中B1矩阵里有n个数值这n个数值的平均值怎么写代码

在TensorFlow 2.6中，可以使用tf.reduce_mean函数来计算一个张量的平均值。假设B1矩阵是一个形状为(n,)的张量，其中包含n个数值，可以按如下方式计算其平均值：

import tensorflow as tf

B1 = tf.Variable([1.0, 2.0, 3.0, 4.0, 5.0])  # 示例B1矩阵

mean = tf.reduce_mean(B1)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    result = sess.run(mean)
    print(result)

输出结果为：3.0

请注意，以上代码使用了tf.Session()来运行计算图，并使用tf.global_variables_initializer()来初始化变量。在TensorFlow 2.0及以后的版本中，也可以使用eager execution模式来计算结果，而无需显式地创建会话。