matmul for each RaggedTensors

Question

I would like to do an tf.matmul for each tensor in RaggedTensors. i tried tf.ragged.map_flat_values(tf.matmul, rt1, rt2, transpose_b=True)(where each rt.shape is sa