在tensorflow中，如何让py_func返回Matrix

Question

我想让我的tf模型火车标记为MNIST数据 简单地说我只想训练一个号码的MNIST 无论如何它并不重要，我需要如何让py_func返回Matrix

<%= link_to(test.name, user_test_result_path(id: protocol.id, user_id: 1)) %>

这是我的功能代码和

def redefine(number):
    pixel, label = MNIST('./mnist/data/').load_training()
    index = 0
    result = list()
    for num in label:
        if num == number:
            result.append(pixel[index])
        index =+ 1
    return result

这是我转换为张量流量的适应值的附加代码 一切都很好，但我不知道我必须填写什么？＆＃39;
因为这不是填写工作

number = tf.placeholder(tf.float32)
tf_redefine = tf.py_func(redefine, [number], ???)

请帮助我们...... 谢谢！