180224 tensorflow之tf.reshape()与tf.concat()应用举例

import tensorflow as tf
I = tf.placeholder(tf.float32,[None,4])
X = np.arange(16).reshape(4,4).astype('float32')
# x = tf.constant(np.arange(4).reshape(1,4).astype('float32'))
W = tf.constant(np.arange(12).reshape(4,3).astype('float32'))
B = tf.constant(np.arange(9).reshape(3,3).astype('float32'))
G = tf.Variable(tf.zeros([3,3]))
# XW = tf.matmul(X,W)
# xW = tf.matmul(x,W)

F = tf.matmul(I,W)
for i in [I,X,W,B,G]:
    print(i)
print('*'*100)
a = []
# code-1
# for i in range(3):
#     t = F+B[i,:]
#     print(t)
#     t = tf.nn.softmax(t)
#     print(t)
#     a.append(t)
# b = tf.concat(a,axis=1)
# c = tf.reshape(b,[-1,3,3])
# code-2
a  = [tf.nn.softmax( F+B[i,:]) for i in range(3)]
c = tf.reshape(a,[-1,3,3])
print(a)
print(c)

with tf.Session() as sess:  
    print(print(sess.run([F],feed_dict={I:X})))
    print('*'*100)
    print(sess.run(B))
    print('*'*100)
    print(sess.run([c],feed_dict={I:X}))