tf.split用法

按照行或者列分割一個矩陣 第一個參數是待切分的張量,第二個是切分的大小,共切分爲三個,每一個分別爲4列,15列,11列。第三個參數爲1表示按照列切分,爲0表示按照行切分。 切分的子塊的大小加起來應該等於母塊的大小。比如這個例子,4+15+11=30列 看下面的代碼例子
相關文章
相關標籤/搜索