numpy-einsum

How to multiply Tensorflow arrays across specified indicies Question: I would like to multiply two Tensorflow Arrays in a certain way as shown in the code below: import tensorflow as tf from tensorflow.keras import mixed_precision policy = mixed_precision.Policy(‘mixed_float16’) mixed_precision.set_global_policy(policy) print(‘Compute dtype: %s’ % policy.compute_dtype) print(‘Variable dtype: %s’ % policy.variable_dtype) a = tf.random.normal(shape=[1000, 1439]) b = …

Understanding NumPy's einsum Question: I’m struggling to understand exactly how einsum works. I’ve looked at the documentation and a few examples, but it’s not seeming to stick. Here’s an example we went over in class: C = np.einsum("ij,jk->ki", A, B) for two arrays: A and B. I think this would take A^T * B, but …