我正在尝试将数组连接在一起。这是我的尝试:
rgb = tf.placeholder(tf.float32, [None, 672, 672, 3])
r, g, b = tf.split(rgb, 3, 3)
bgr = tf.stack(
[
b-23.5,
g-30.6,
r-40.1
],3)
print "rgb :", rgb
print "r :",r
print "bgr :", bgr
但是,bgr
的形状不正确。而不是(?, 672, 672, 3, 1)
我想要与rgb相同的形状,即(?, 672, 672, 3)
rgb : Tensor("mul:0", shape=(?, 672, 672, 3), dtype=float32)
r : Tensor("split:0", shape=(?, 672, 672, 1), dtype=float32)
bgr : Tensor("stack:0", shape=(?, 672, 672, 3, 1), dtype=float32)
答案 0 :(得分:2)
尝试{{3}}
bgr = tf.concat([b-23.5,
g-30.6,
r-40.1], axis=3)
它在维度axis
上连接张量列表而不引入新维度。要连接的张量应该具有相同的形状,除了它们连接在一起的维axis
。