tfl.lattice_lib.batch_outer_operation

View source on GitHub

Computes outer operation of last dimensions of each of given tensors.

tfl.lattice_lib.batch_outer_operation(
    list_of_tensors,
    operation=tf.multiply
)

Args:

• list_of_tensors: List of tensors of same shape (batch_size, ..., k[i]) where everything expect k_i matches.
• operation: Binary TF operation which supports broadcasting to be applied.

Returns:

Tensor of shape: (batch_size, ..., mul_i(k[i])).