在操作矩阵的时候,不同的接口对于矩阵的输入维度要求不同,输入可能为1-D,2-D,3-D等等。下面介绍一下使用Numpy进行矩阵维度变更的相关方法。主要包括以下几种:1、np.newaxis扩充矩阵维度2、np.expand_dims扩充矩阵维度3、np.squeeze删除矩阵中维度大小为1的维度np.newaxis,np.expand_dims扩充矩阵维度:importnumpyasnpx=np.arange(8).reshape(2,4)print(x.shape)#添加第0维,输出shape->(1,2,4)x1=x[np.newaxi...
继续阅读 >
分类:矩阵维度
2020
09-29
09-29
Python3 Tensorlfow:增加或者减小矩阵维度的实现
1.增加维度下面给出两个样例样例1:[1,2,3]==>[[1],[2],[3]]importtensorflowastfa=tf.constant([1,2,3])b=tf.expand_dims(a,1)withtf.Session()assess:a_,b_=sess.run([a,b])print('a:')print(a_)print('b:')print(b_)输出结果a:[123]b:[[1][2][3]]样例2:[1,2,3]==>[[1,2,3]]importtensorflowastfa=tf.constant([1,2,3])b=tf.expand_dims(a,0)withtf.Session()assess:a_,b_=se...
继续阅读 >