NumPy: найти (непрерывные) вхождения подматрицы ⇐ Python

[Anonymous]

У меня есть матрица Y и матрица A с размерами меньше, чем у Y. Как в numpy быстро найти место, где A встречается как непрерывная подматрица Y (скажем, индекс в верхнем левом углу)?
Например, если у меня

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

A = np.array([[1, 0],
[0, 1]])
Y = np.array([[0, 1, 0],
[0, 0, 1],
[1, 0, 0]])

Тогда хотелось бы, чтобы результат был

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

[[0, 1]]

поскольку A встречается как подматрица, начинающаяся с первой строки и второго столбца.
Я знаю, что это было бы легко реализовать с помощью циклы, но мне бы хотелось что-то, что работает быстрее.

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