如何在Tensorflow中使用张量生成数据集

Question

我需要一种数据集张量，shape = [160000，1，2]，我想生成 组合如下，但从0到400

[[[0 0]]

[[0 1]]

[[0 2]]

[[0 3]]

[[1 0]]

[[1 1]]

[[1 2]]

[[1 3]]

[[2 0]]

[[2 1]]

[[2 2]]

[[2 3]]]

Answer 1

首先创建一个numpy数组，并使用它创建张量。以下代码输出张量shape = [160000, 1, 2]，其中每个值都在0 to 400之间。

import numpy as np
import tensorflow as tf
arr = np.random.random_integers(low=0, high=400, size=(160000, 1, 2))
tensor = tf.constant(arr)