La matriz de similitud de dígitos

Aplana los 1797 dígitos manuscritos reales a (1797, 64) y calcula la similitud entre cada par con un solo einsum:

np.einsum('id,jd->ij', D, D)    # (1797, 1797)

d es el índice de píxel y se contrae; i y j sobreviven ambos, uno por dígito, y por eso el resultado es cuadrado. Son 3.2 millones de productos punto desde una sola línea, y es exactamente la operación que ejecuta un recomendador para ordenar artículos para un usuario.