tensorflow中的矩阵转置
import tensorflow as tf raw = tf.Variable(tf.random_normal(shape=(4,3,2)))#4个数组,3行两列 a = tf.transpose(raw,perm=[2,0,1])#两个数组,4行3列 b = tf.transpose(raw,perm=[1,0,2])#3个数组,4行2列 with tf.Session() as...
原创
2018-09-06 20:57:36 ·
1510 阅读 ·
0 评论