En DGEMM http://www.netlib.org/lapack/explore-html/d1/d54/groupdoubleblas_level3gaeda3cbd99c8fb834a60a6412878226e1.html#gaeda3cbd99c8fb834a60a6412878226e1 se puede realizar C = alpha * A.B + beta*C En eimsum, hay ij,jk->ik (o índices más generales) para A.B. ¿Existe alguna forma eficiente de incorporar alpha y beta*C, presumiblemente si A y B son arreglos de rango general? ¿O debo escribir varias líneas de operaciones con . . . Read more
Soy nuevo en Einsum y quería un caso particular: usar einsum para multiplicar todos los elementos de una matriz entre sí; digamos que se tiene una matriz 2D: py np.random.rand((16,2)) Multiplicar los elementos a lo largo de un eje obteniendo (16,) y luego multiplicarlos entre sí nuevamente para obtener un . . . Read more
Considere el siguiente fragmento de código de PyTorch: X = torch.einsum(“rij, sij -> rs”, A, A) Y = torch.einsum(“rij, sij -> rs”, B, B) Z = torch.einsum(“rij, sij -> rs”, C, C) torch.einsum(“ij, ij, ij -> “, X, Y, Z) El cual realiza la siguiente suma: ¿Es posible formular esto . . . Read more