如何在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)