我有这样的张量：
tensor_a = [[[[255,255,255]]], [[[100,100,100]]]]
tensor_b = [[[[0.1,0.2]]], [[[0.3,0.4]]]]
tensor_c = [[[[1]]], [[[2]]]]
今天我尝试将上面的这些张量连接到 tensor_d 像：
tensor_d = [[[[255,255,255,0.1,1]]], [[[100,100,100, 0.3, 2]]]]
但我不知道如何连接它们。
我曾尝试使用 for 循环将张量附加到列表
但这太慢了（在 tensor_a:(10,64,64,3) 的形状下）
您可以使用张量操作，例如tf.split和tf.concat。
import tensorflow as tf
# tensors
tensor_a = [[[[255, 255, 255]]], [[[100, 100, 100]]]]
tensor_b = [[[[0.1, 0.2]]], [[[0.3, 0.4]]]]
tensor_c = [[[[1]]], [[[2]]]]
# casting becuase date type should match in tf.concat
tensor_a = tf.cast(tensor_a, dtype=tf.float32)
tensor_c = tf.cast(tensor_c, dtype=tf.float32)
# split elements into one and the other at the last axis
b, _ = tf.split(value=tensor_b, num_or_size_splits=[1, -1], axis=-1)
c, _ = tf.split(value=tensor_c, num_or_size_splits=[1, -1], axis=-1)
# concatenate tensors at the last axis
tensors = tf.concat(values=[tensor_a, b, c], axis=-1)
sess = tf.Session()
result = sess.run(tensors)
print(result)
[[[[2.55e+02 2.55e+02 2.55e+02 1.00e-01 1.00e+00]]]
[[[1.00e+02 1.00e+02 1.00e+02 3.00e-01 2.00e+00]]]]
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句