1. 程式人生 > >tf.reshape()的用法

tf.reshape()的用法

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

函式的作用是將tensor變換為引數shape的形式。其中shape為一個列表形式,特殊的一點是列表中可以存在-1。

-1代表的含義是不用我們自己指定這一維的大小,函式會自動計算,但列表中只能存在一個-1.

shape可以看著是reshape變換後的矩陣大小,先不管-1的那一個維度,先看其他維度,然後用原矩陣的總元素個數除以確定的維度,就能得到-1維度的值。