10六、TensorFlow變量 (二) reshape

import tensorflow as tf rank_three_tensor = tf.ones([3, 4, 5]) # 建立一個[3,4,5]大小的張量,3行4列,每一個位置上有五個元素 matrix = tf.reshape(rank_three_tensor, [6, 10]) # 將當前變量reshape成[6,10]個大小的變量 matrixB
相關文章
相關標籤/搜索