softmax output
remove downsampling
change according to changes in rnn_model and loc_data_gen
symmetric S constraint
each datum sum to 1
save A and d to state