(pair_a, pair_b)
| 347 | |
| 348 | @def_function.function() |
| 349 | def pairs_mul(pair_a, pair_b): |
| 350 | return pair(matmul(pair_a.a, pair_b.a), matmul(pair_a.b, pair_b.b)) |
| 351 | |
| 352 | a = constant_op.constant([[1.0, 2.0], [1.0, 2.0]]) |
| 353 | b = constant_op.constant([[3.0, 4.0], [3.0, 4.0]]) |