急求！Python3.5下运行报错

我用的Python3.5，TensorFlow0.11.0

def gen_poetry():  
    def to_word(weights):  
        t = np.cumsum(weights)  
        s = np.sum(weights)  
        sample = int(np.searchsorted(t, np.random.rand(1)*s))  
        return words[sample]  

    _, last_state, probs, cell, initial_state = neural_network()  

    with tf.Session() as sess:  
        sess.run(tf.initialize_all_variables())  

        saver = tf.train.Saver(tf.all_variables())  
        saver.restore(sess, 'poetry.module-49')  

        state_ = sess.run(cell.zero_state(1, tf.float32))  

        x = np.array([list(map(word_num_map.get, '['))])  
        [probs_, state_] = sess.run([probs, last_state], feed_dict={input_data: x, initial_state: state_})  
        #word = to_word(probs_)  
        word = words[np.argmax(probs_)]  
        poem = ''  
        while word != ']':  
            poem += word  
            x = np.zeros((1,1))  
            x[0,0] = word_num_map[word]  
            [probs_, state_] = sess.run([probs, last_state], feed_dict={input_data: x, initial_state: state_})  
            #word = to_word(probs_)  
            word = words[np.argmax(probs_)]  
        return poem  

print(gen_poetry()) 

sample = int(np.searchsorted(t, np.random.rand(1)*s))

return words[sample]

你打印一下sample，和words，应该是sample的值超过words的范围了，所以越界错误了