Tensorflow里的reshape函数

来源:互联网 发布:南开大学VPN网络 编辑:程序博客网 时间:2024/06/05 14:56

API里函数为:

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

作用是将tensor变换为shape的形式

其中shape为一个列表形式,特殊的一点是列表中可以存在-1。-1代表的含义是不用我们自己指定这一维的大小,函数会自动计算,但列表中只能存在一个-1。

变换过程:首先将tensor变换为以为矩阵,然后再变成shape形状

实例如下:


第一次变换 =》 [1,2,3,4]

第二次变换 =》 [[1],[2],[3],[4]]