tensorflow 不同维度的 tensor 如何合并?

我的 scv 文件中每行一个样本,假设一个样本只有两个特征 x1, x2。
reader 按行读之后可以用 tf.stack 合并成 [x1, x2]
现在想要将所有的样本合并成一个 m*2 的 tensor,也就是这样:
[[x1^(1), x2^(1)]
[x1^(2), x2^(2)]
...
[x1^(m), x2^(m)]]
但是不同维度的 tensor 好像无法合并..
比如:
[[x1^(1), x2^(1)]
[x1^(2), x2^(2)]]

[[x1^(3), x2^(3)]]
合并会报错:
From merging shape 0 with other shapes. for 'stack_1' (op: 'Pack') with input shapes: [2,2], [1,2].
求解...

阅读 10.6k
1 个回答

新建axis时有用stack,在原有的某axis上拼接用concat

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题
宣传栏