将一个维度为4 256 256 3的tensor转换为4 256 256

可以使用tf.reduce_mean函数来实现：

import tensorflow as tf

# 假设input_tensor是一个维度为[4, 256, 256, 3]的tensor
input_tensor = tf.placeholder(tf.float32, [4, 256, 256, 3])

# 将最后一个维度上的值取平均，得到一个维度为[4, 256, 256]的tensor
output_tensor = tf.reduce_mean(input_tensor, axis=-1)

也可以使用tf.squeeze函数来实现：

import tensorflow as tf

# 假设input_tensor是一个维度为[4, 256, 256, 3]的tensor
input_tensor = tf.placeholder(tf.float32, [4, 256, 256, 3])

# 将最后一个维度上的值去掉，得到一个维度为[4, 256, 256]的tensor
output_tensor = tf.squeeze(input_tensor, axis=-1)

两种方法都可以将一个维度为[4, 256, 256, 3]的tensor转换为[4, 256, 256]的tensor