This page cannot be shown here. You can still read it on the original site — the toolbar below keeps your place in the directory.
The einsum function in numpy , jax or pytorch is a compact way of describing tensor operations which involve pairing up values, multiplying the pairs, and summing their products. The two operations dot and matmul can be written in einsum notation as i,i-> and mn,no->mo respectively. These strings assign letters to the dimensions of each input and output. Repeated letters from the input must match…
Comments
Nothing yet. Say the first thing.
Sign in to join the conversation.