it is a good job , in the file stateful_modules.py,there is a assign code as follow:
self.update_h = h_state.assign(tf.expand_dims(inter_LSTM_h,axis = 0),read_value = False)
the h_state shape is (b,f,c) ,the parentheses part's shape is （1,b*f,c）,why can be assign to h_state, the shape is not equal?
iknow why in this place b =1 so the shape is the same
it is a good job , in the file stateful_modules.py,there is a assign code as follow:
self.update_h = h_state.assign(tf.expand_dims(inter_LSTM_h,axis = 0),read_value = False)the h_state shape is (b,f,c) ,the parentheses part's shape is （1,b*f,c）,why can be assign to h_state, the shape is not equal?
iknow why in this place b =1 so the shape is the same