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

发布于 2022-09-06 08:04:05 字数 422 浏览 10 评论 0

我的 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].
求解...

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

时光暖心i 2022-09-13 08:04:05

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

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文