GitHub

tfl.lattice_lib.batch_outer_operation

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

tfl.lattice_lib.batch_outer_operation(
    list_of_tensors, operation='auto'
)

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

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

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

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-02-19 UTC.