0
点赞
收藏
分享

微信扫一扫

tensorflow 除法


import tensorflow as tf

a1 = tf.constant([[1,2],[3,4],[5,6]])
a2 = tf.constant([[[1],[2]],[[3],[4]],[[5],[6]]])
a3 = tf.constant([[[[1]],[[2]]],[[[3]],[[4]]],[[[5]],[[6]]]])
r1 = a1/tf.constant(2)
r2 = a2/tf.constant(2)
r3 = a3/tf.constant(2)
sess = tf.Session()
print(sess.run(r1))
print(sess.run(r2))
print(sess.run(r3))

[[ 0.5 1. ]
[ 1.5 2. ]
[ 2.5 3. ]]

[[[ 0.5]
[ 1. ]]
[[ 1.5]
[ 2. ]]
[[ 2.5]
[ 3. ]]]

[[[[ 0.5]]
[[ 1. ]]]
[[[ 1.5]]
[[ 2. ]]]
[[[ 2.5]]
[[ 3. ]]]]

也就是说无论多少维的tensor,都是对于最终的每个元素都除的


举报

相关推荐

试除法

矩阵的除法

java 计算除法

高精度除法

模拟除法过程

辗转相除法

lua除法bug

0 条评论