conve_interaction

conve_interaction(h, r, t, t_bias, input_channels, embedding_height, embedding_width, hr2d, hr1d)

Evaluate the ConvE interaction function.

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.

• t_bias (FloatTensor) – shape: (*batch_dims) The tail entity bias.

• input_channels (int) – The number of input channels.

• embedding_height (int) – The height of the reshaped embedding.

• embedding_width (int) – The width of the reshaped embedding.

• hr2d (Module) – The first module, transforming the 2D stacked head-relation “image”.

• hr1d (Module) – The second module, transforming the 1D flattened output of the 2D module.

Return type:

FloatTensor

Returns:

shape: batch_dims The scores.