Как сделать ту же функцию для pytorch expand_as в керасе

Question

код pytorch:

att_h = att_h.unsqueeze(1).expand_as(att)

att_h форма - (10,512)

att форма - (10,196,512)

Keras код:

K.expand_dims(att_h, 1).expand_as(att)

Произошла ошибка: объект 'Tensor' не имеет атрибута 'expand_as'

Не уверен, как сделать то же самое в кератах.

Answer 1

В Keras нет встроенной функции как таковой.Но вы можете достичь того же результата, используя: np.reshape.

Так что вы можете:

K.expand_dims(att_h,1).reshape(att.shape)