Python Numpy скалярное целое число, преобразованное

Question

Я получил сообщение об ошибке по этому коду:

import numpy as np

def transform(X, a=1):
    """
    param X: np.array[batch_size, n]
    """
    for i in range(np.size(X, 0)):
        out = np.copy(X)
        x = out[i]
        tmp = np.copy(x)
        for j in range(x.size):
            if (j % 2 == 1):
                tmp[j] = a
            else:
                tmp[j] = tmp[j]**3
        out[i] = np.concatenate(x, np.flip(tmp, 0), None)
    return out

Сообщение об ошибке: введите описание изображения здесь

Как я могу исправить эту функцию. Ожидается, что X - это двумерный массив (массив массивов?) X-некоторая строка (массив-элемент) X. Это пример того, что моя программа должна делать с каждой строкой X: введите описание изображения здесь