Quite likely, one could use some index trickery, but what is a good way to achieve things really depends on what realistic shapes are.
So A is a list of … tensors of shape … to … or so, each multiplied with a separate scalar.