TransFInteraction

class TransFInteraction(*args, **kwargs)[source]

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A stateless module for the TransF interaction function.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

func(r, t)

Evaluate the TransF interaction function.

Methods Documentation

func(r, t)

Evaluate the TransF 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.

Return type:

FloatTensor

Returns:

shape: batch_dims The scores.