Is there a way to do einsum in Python with boolean logic to optimize it?

Question

Okay, so basically i\'m working on some sort of image blending thing, and i have a function that will blend every image in the given array according to some specified weight, li