dist_ma_interaction

dist_ma_interaction(h, r, t)

Evaluate the DistMA interaction function from [shi2019].

\[\langle h, r\rangle + \langle r, t\rangle + \langle h, t\rangle\]

Parameters:

• h (FloatTensor) – shape: (*batch_dims, dim) The head representations.

• r (FloatTensor) – shape: (*batch_dims, dim) The relation representations.

• t (FloatTensor) – shape: (*batch_dims, dim) The tail representations.

Return type:

FloatTensor

Returns:

shape: batch_dims The scores.