时间戳

首页 > 解决方案 > 附加 tf.ones 和 tf.zeros（张量流）

python - 附加 tf.ones 和 tf.zeros（张量流）

问题描述

我的愚蠢代码：

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.zeros 代替。

但我不知道如何将两个张量附加在一起。我知道我可以评估结果以取回一个 numpy 数组。但是如何附加它们？

标签： pythonclasstensorflowclassification

解决方案

嗯嗯嗯嗯嗯。我讨厌不得不回答我自己的问题。

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()

推荐阅读