tensorflow中的reshape(tensor,[1,-1])和reshape(tensor,[-1,1])
tf.reshape(tensor,[-1,1])将张量变为一维列向量
tf.reshape(tensor,[1,-1])将张量变为一维行向量
import tensorflow as tf
a = [[1,2],[3,4],[5,6]]
tf.reshape(a,[-1,1])
Out[13]: <tf.Tensor 'Reshape_4:0' shape=(6, 1) ...
原创
2019-11-23 17:26:36 ·
1345 阅读 ·
0 评论