1、tf.shape(A) # 獲取張量A（陣列，list, tensor張量）的大小，返回的是一個list。

import tensorflow as tf
import numpy as np

a_array=np.array([[1,2,3],[4,5,6]])
b_list=[[1,2,3],[3,4,5]]
c_tensor=tf.constant([[1,2,3],[4,5,6]])

with tf.Session() as sess:
    print(sess.run(tf.shape(a_array)))
    print(sess.run(tf.shape(b_list)))
    print(sess.run(tf.shape(c_tensor)))
 

返回[2, 3]，[2, 3]，[2, 3]。

2、x.get_shape().as_list()

x.get_shape()，只有tensor才可以使用這種方法，返回的是一個元組

import tensorflow as tf
import numpy as np

a_array=np.array([[1,2,3],[4,5,6]])
b_list=[[1,2,3],[3,4,5]]
c_tensor=tf.constant([[1,2,3],[4,5,6]])

print(c_tensor.get_shape())
print(c_tensor.get_shape().as_list())

with tf.Session() as sess:
    print(sess.run(tf.shape(a_array)))
    print(sess.run(tf.shape(b_list)))
    print(sess.run(tf.shape(c_tensor)))
 

返回：（2, 3）,[2, 3],  [2, 3], [2, 3],  [2, 3]

只能用於tensor來返回shape，但是是一個元組，需要通過as_list()的操作轉換成list.

3、tf.split()

tf.split(dimension, num_split, input)：dimension的意思就是輸入張量的哪一個維度，如果是0就表示對第0維度進行切割。num_split就是切割的數量，如果是2就表示輸入張量被切成2份，每一份是一個列表。

import tensorflow as tf
import numpy as np

A = [[1,2,3],[4,5,6]]
x = tf.split(1, 3, A)

with tf.Session() as sess:
	c = sess.run(x)
	for ele in c:
		print ele

輸出： 

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