Isn’t that expected? You are comparing one tensor multiplication with multiple ones. While they are equivalent, the first have can have better optimization basing on size and less overhead. It is natural that the second is much slower.