追加tf.ones公司以及tf.0（张力流）

Question

我愚蠢的代码：

ones = [1 for i in range(len(positivePictures))]
zeros = [0 for i in range(len(negativePictuers))]
y = zeros + ones
classes = np.array(y).reshape(1,len(y))

这是可行的，但它的时间太长（约4k照片）。我在考虑用

在tf.ones公司以及tf.0相反。在

但我不知道怎么把两个张量加在一起。我知道我可以计算结果来得到一个numpy数组。但是如何附加它们呢？在

Answer 1

啊啊。 我讨厌自己回答问题。在

def getOnesAndZeros(num_ones, num_zeros):

        sess = tf.Session()
        with sess.as_default():
            ones = tf.ones(
                (num_ones, 1),
                dtype=tf.float32,
                name=None
            )
            zeros = tf.zeros(
                (num_zeros, 1),
                dtype=tf.float32,
                name=None
            )

            both = tf.concat([ones, zeros], 0)
            npBoth = both.eval()
            return npBoth.flatten()