tf.transpose

来源:互联网 发布:软件开发管理名词解释 编辑:程序博客网 时间:2024/06/10 04:49

tensorflow 里面的 tensor是先从高维向低维算起的 
比如:’x’ is [[[1 2 3] 
[4 5 6]] 
[[7 8 9] 
[10 11 12]]] 
x的维数就是[2,2,3] 
而perm的意思就是将tensor对应的不同的维数之间变换 
比如perm = [2,1,0],则原tensor = [3, 2,2],全部倒过来 
perm = [0,2,1],则原tensor = [2,3,2], 后两维置换

tf.transpose(x, perm=[0, 2, 1]) ==> 
[[[1 4] 
[2 5] 
[3 6]] 
[[7 10] 
[8 11] 
[9 12]]]