Подтвердить что ты не робот

Поменяйте оси тензоров тензометра

У меня тензор формы (30,116,10), и я хочу поменять первые два десятилетия, так что у меня есть тензор формы (116,30,10)

Я видел, что numpy как таковая функция реализована (np.swapaxes), и я искал что-то подобное в tensorflow, но ничего не нашел.

Есть ли у вас какие-либо идеи?

Заранее спасибо,

Alexis

Ответ 1

tf.transpose предоставляет ту же функциональность, что и np.swapaxes, хотя и в более обобщенной форме. В вашем случае вы можете сделать tf.transpose(orig_tensor, [1, 0, 2]) который будет эквивалентен np.swapaxes(orig_np_array, 0, 1).