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='auto'
)

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.
  • string "auto" in order to apply tf.multiply for first several tensors and tf.matmul for remaining.

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