如何在TensorFlow中交换张量轴?
tensorflow
10
0

我有一个形状为(30, 116, 10) 30,116,10)的张量,我想交换前两个维,以便我有一个形状为(116, 30, 10)的张量

我看到numpy实现了这样的功能( np.swapaxes ),我在张量流中搜索了类似的内容,但没有发现任何东西。

你有什么主意吗?

参考资料:
Stack Overflow
收藏
评论
共 1 个回答
高赞 时间 活跃

tf.transpose提供相同的功能np.swapaxes ,虽然在一个更广义的形式。在您的情况下,您可以执行tf.transpose(orig_tensor, [1, 0, 2]) ,它等效于np.swapaxes(orig_np_array, 0, 1)

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题