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

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

Answer 1

UHHHHHG。我讨厌回答我自己的问题。

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

追加tf.ones和tf.zeros（tensorflow）

1 个答案: