tensorflow 2.0 で足し算をしたい

Question

最初にコードを失礼します

import tensorflow as tf

v1 = tf.placeholder(tf.int32)
v2 = tf.placeholder(tf.int32)
add = v1 + v2

session = tf.Session()
print(session.run(add, {v1: 3, v2: 5}))  # => 8
print(session.run(add, {v1: [1, 2], v2: [3, 4]}))  # => [4, 6]

上に示したコードをtensorflow 2.0 に置き換えたく下のコードに書き換えましたが、リストの処理が要素ごとの足し算ではなく、連結になってしまいます。
できればひとつの関数で単一の数字とリストの両方とも処理できるようにしたいです。
少しでもアドバイスございましたら、ご教示お願いします。

import tensorflow as tf

@tf.function
def add(v1, v2):
    return v1 + v2

tf.print(add(3, 5)) # => 8
tf.print(add([1, 2], [3, 4])) # => [1, 2, 3, 4]

Answer 1

tf.math.addが使えます。

import tensorflow as tf

@tf.function
def add(v1, v2):
    return tf.math.add(v1, v2)

tf.print(add(3, 5)) # => 8
tf.print(add([1, 2], [3, 4])) # => [4 6]

Comments

[UNICO.SA]

ご回答ありがとうございます。
基礎からしっかりと知識をつけていきます。