Python tensorflow切片尺寸,仅更改张量的一个通道

Python tensorflow切片尺寸,仅更改张量的一个通道,python,tensorflow,Python,Tensorflow,我想把一个通道乘以2。我尝试了下面的代码,但不起作用,有人知道为什么吗 preds = tf.add_n(bases, name="output") output_list = [] for i in range(3): if i == 0: output_list.append(tf.multiply(preds[:,:,:,i],2)) output_list.append(preds[:,:,:,i]) preds = tf.stack(output_lis

我想把一个通道乘以2。我尝试了下面的代码,但不起作用,有人知道为什么吗

preds = tf.add_n(bases, name="output")
output_list = []
for i in range(3):
    if i == 0:
        output_list.append(tf.multiply(preds[:,:,:,i],2))
    output_list.append(preds[:,:,:,i])
preds = tf.stack(output_list)

嗨,欢迎来到苏。您得到的输出是什么?还是一个错误?请在你的问题中展示给其他人,以便更好地帮助你!代码中一个可能的错误是没有
else
子句,因此第一层将添加两次