How do I swap tensor's axes in TensorFlow?

后端 未结 2 1984
广开言路
广开言路 2020-12-05 06:59

I have a tensor of shape (30, 116, 10), and I want to swap the first two dimensions, so that I have a tensor of shape (116, 30, 10)

I saw t

2条回答
  •  轻奢々
    轻奢々 (楼主)
    2020-12-05 07:40

    tf.transpose provides the same functionality as np.swapaxes, although in a more generalized form. In your case, you can do tf.transpose(orig_tensor, [1, 0, 2]) which would be equivalent to np.swapaxes(orig_np_array, 0, 1).

提交回复
热议问题