Поменяйте оси тензоров тензометра
У меня тензор формы (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)
.