Эдвард KLqp: Пара ключ-значение в данных не имеет одинаковую форму: (?,), (?, 1)

Question

Я новичок в Эдварде. Я создал два ed.RandomVariable точно такого же размера, но я получаю TypeError: Key-value pair in data does not have same shape: (?,), (?, 1),

Код, который я запускаю

u = Normal(loc=tf.zeros([num_samples, u_dim]), 
           scale=tf.ones([num_samples, u_dim]), name='gaussian_pu')
qu = Normal(loc=mu_u, scale=sigma_u, name='gaussian_pu_amlry')
inference = ed.KLqp({u: qu}, data)

Когда я печатаю u,qu, Я получил

(<ed.RandomVariable 'gaussian_pu/' shape=(?, 20) dtype=float32>,
 <ed.RandomVariable 'gaussian_pu_amlry/' shape=(?, 20) dtype=float32>)

который, кажется, имеет точно такое же измерение...