tensorflow用自己的数据创建batch

使用自己的数据创建batch

下面以一个例子说明：

data = np.array([x for x in range(1, 101)]).reshape(10, 10)
label = np.array([x for x in range(1, 11)]).reshape(10, 1)

将data代表的矩阵作为输入的数据，其中每一行作为一个样例，每一列作为一个特征，这里要说明的是：如果你的数据是2维的数据完全可以存为这种格式，就像是mnist集中的数据也是将每一张图片flatten后存为一维的行向量。

附上所有代码：

import numpy as np
import tensorflow as tf

...

if __name__ == '__main__':
    data = np.array([x for x in range(1, 101)]).reshape(10, 10)
    label = np.array([x for x in range(1, 11)]).reshape(10, 1)
    batch_size = 3
    capacity &