tensorflow + tf.transposs

来源:互联网 发布:苹果离线翻译软件 编辑:程序博客网 时间:2024/05/17 22:22

今天使用tensorflow遇到一个函数 tf.transpose(X,perm = [0,2,1])

这个函数就是将tensor的对应的维数进行置换,这里发现几点,

1.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(b, perm=[0, 2, 1]) ==>

                                      [[[1  4]
                                        [2  5]
                                        [3  6]]

                                       [[7 10]
                                        [8 11]
                                        [9 12]]] 


0 0
原创粉丝点击