attn_for_self = tf.einsum(“…NHI , IHO → …NHO”, x, self.attn_kernel_self) Does ND4J have einsum function like it in TF?