テンソルフローまたはケラールなしでの形状変更

Question

テンソル形状が[None, 80, 80]であるとします。これは、確率的勾配降下のための80x80画像のバッチです。

minibatchのサイズを50（Noneは50）とし、Noneを2つの次元に分解して(5, 10)とすると、[?, ?, 80, 80]となります。

グラフをNone値で作成するにはどうすればよいですか？

Answer 1

あなたはtf.reshapeでそれを行うことができます。

import numpy as np 
import tensorflow as tf 

x = tf.placeholder(tf.float32, shape=[None, 80, 80], name='x') 
y = tf.reshape(x, shape=[-1, 10, 80, 80], name='y') 
data = np.zeros([50, 80, 80]) 
with tf.Session() as session: 
    result = session.run(y, feed_dict={x: data}) 
    print result.shape 

結果出力：もちろん

(5, 10, 80, 80) 

を、不適当バッチサイズを渡すと、実行時に例外が発生しますことを心に留めておきます。