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

在TensorFlow 2.6中，可以使用tf.reduce_mean函数来计算矩阵中数值的平均值。以下是一个示例代码：

import tensorflow as tf

# 假设B1矩阵是一个1维张量
B1 = tf.constant([1, 2, 3, 4, 5], dtype=tf.float32)

# 计算平均值
mean_value = tf.reduce_mean(B1)

# 打印结果
print(mean_value.numpy())

这个代码将计算B1矩阵中数值的平均值，并打印结果。请确保你的B1矩阵是一个张量，并且其中的数值类型为tf.float32或tf.float64，因为tf.reduce_mean函数只支持这些数值类型。如果你的B1矩阵不是1维的，请在使用tf.reduce_mean函数之前先对其进行展平操作。另外，请确保你已经正确导入了tensorflow库。