можем ли мы объединить более двух данных в один тензор

Question

У меня есть три массива numpy, содержащих мои данные.

X_train = np.zeros((1, 288, 288, 3), dtype=np.uint8)

X_train2 = np.zeros((1, 288, 288, 3), dtype=np.uint8)

X_train3 = np.zeros((1, 288, 288, 3), dtype=np.uint8)

Используя np.concatenate, я могу объединить два изображения в один тензор, как показано ниже:

X_train2=np.concatenate([X_train, X_train2], axis = -1)

Я хочу объединить несколько изображений X_train, X_train2 и X_train3 в один тензор, можно ли использовать numpy.

Answer 1

Да. Вы можете объединить столько массивов numpy, сколько захотите.

X_train_final = np.concatenate([X_train, X_train2, X_train3], axis = -1)

допустимо и даст вам массив, в котором последнее измерение будет в 3 раза больше, чем в исходном массиве. Вы можете продолжить этот способ для любого количества массивов.

Прочтите документацию: https://numpy.org/doc/1.18/reference/generated/numpy.concatenate.html