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
相关文章
相关标签/搜索