tf.reshape()的用法

tf.reshape(tensor, shape, name=None)

函数的作用是将tensor变换为参数shape的形式。其中shape为一个列表形式,特殊的一点是列表中可以存在-1。

-1代表的含义是不用我们自己指定这一维的大小,函数会自动计算,但列表中只能存在一个-1.

shape可以看着是reshape变换后的矩阵大小,先不管-1的那一个维度,先看其他维度,然后用原矩阵的总元素个数除以确定的维度,就能得到-1维度的值。

 

猜你喜欢

转载自blog.csdn.net/gqixf/article/details/83093691