Странное взаимодействие между изменением формы и приведением типов в Numba ⇐ Python

[Anonymous]

Я заметил, что когда я передаю двумерный массив из 0 и 1 в функцию Numba njit, изменяю его форму, а затем привожу к np.int32 или numba.int32, результирующий массив при печати отличается. >
Вот пример кода:

Код: Выделить всё

import numpy as np
from numba import njit

array_2d = np.array([[0, 1, 0, 1, 1, 0, 0, 1],
[0, 1, 1, 0, 1, 1, 0, 0],
[0, 1, 1, 0, 1, 1, 0, 0]]).T

num_cols = array_2d.shape[1]
num_rows = array_2d.shape[0]

@njit
def f(array, num_rows, num_cols):
pairs = array.reshape(num_rows // 2, 2, num_cols)

pairs_cast = pairs.astype(numba.int32)

return pairs, pairs_cast

pairs, pairs_cast = f(array_2d, num_rows, num_cols)

print("Pairs:")
print(pairs)
print("\nPairs cast to int32:")
print(pairs_cast)

Вывод:

Код: Выделить всё

Pairs:
[[[0 0 0]
[1 1 1]]

[[0 1 1]
[1 0 0]]

[[1 1 1]
[0 1 1]]

[[0 0 0]
[1 0 0]]]

Pairs cast to int32:
[[[0 0 0]
[1 1 1]]

[[1 1 1]
[0 1 1]]

[[0 1 1]
[0 0 0]]

[[1 0 0]
[1 0 0]]]

Было бы любопытно узнать, что здесь происходит.


Подробнее здесь: https://stackoverflow.com/questions/791 ... g-in-numba