tensorflow常用函数解析

Posted 2020-10-14 光彩照人

一、tf.transpose函数的用法

tf.transpose(input, [dimension_1, dimenaion_2,..,dimension_n]):这个函数主要适用于交换输入张量的不同维度用的，如果输入张量是二维，就相当是转置。dimension_n是整数，如果张量是三维，就是用0,1,2来表示。这个列表里的每个数对应相应的维度。如果是[2,1,0]，就把输入张量的第三维度和第一维度交换。

import numpy as np
import tensorflow as tf

A = np.array([[1,2,3],[4,5,6]])
x = tf.transpose(A, [1,0])

B = np.array([[[1,2,3],[4,5,6]]])
y = tf.transpose(B, [2,1,0])

with tf.Session() as sess:
    print("A:\n",A)
    print("new_A:\n",sess.run(x))
    print("B:\n",B)
    print("new_B:\n",sess.run(y))

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

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

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

new_B:
[[[1]
  [4]]

[[2]
  [5]]

[[3]
  [6]]]

分析理解：对于第一个二维矩阵A来说就比较简单了，就是普通的转置。对于第二个三维变量B，在转置前可以理解为是1片2*3的矩阵，而转置操作要求第一维和第三维置换，那就是要变成3片2*1的矩阵，这样就好理解了，输出结果正是3片1*2的矩阵。