Как создать набор данных, используя тензор в Tensorflow

Question

Мне нужен тензор вида Dataset, 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

Сначала создайте пустой массив и создайте тензор, используя его.Следующий код выводит тензор 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)