deepr.examples.multiply.layers package

Submodules

deepr.examples.multiply.layers.loss module

Squared L2 loss

class deepr.examples.multiply.layers.loss.SquaredL2

Bases: Layer

forward(tensors, mode: Optional[str] = None)

Forward method on one Tensor or a tuple of Tensors.

Parameters:

• tensors (Union[tf.Tensor, Tuple[tf.Tensor, ...]]) –

  • n_in = 1: one tensor (NOT wrapped in a tuple)

  • n_in > 1: a tuple of tensors

• mode (str, optional) – Description

Returns:

• n_out = 1: one tensor (NOT wrapped in a tuple)

• n_out > 1: a tuple of tensors

Return type:

Union[tf.Tensor, Tuple[tf.Tensor, …]]

deepr.examples.multiply.layers.model module

Learned Multiply Model

class deepr.examples.multiply.layers.model.Multiply

Bases: Layer

forward(tensors, mode: Optional[str] = None)

Forward method on one Tensor or a tuple of Tensors.

Parameters:

• tensors (Union[tf.Tensor, Tuple[tf.Tensor, ...]]) –

  • n_in = 1: one tensor (NOT wrapped in a tuple)

  • n_in > 1: a tuple of tensors

• mode (str, optional) – Description

Returns:

• n_out = 1: one tensor (NOT wrapped in a tuple)

• n_out > 1: a tuple of tensors

Return type:

Union[tf.Tensor, Tuple[tf.Tensor, …]]

Module contents