当前位置:网站首页>TF中的四则运算

TF中的四则运算

2022-08-11 06:45:00 phac123

只有维度相同的张量才可以做四则运算

import tensorflow as tf

a = tf.ones([1, 3])
b = tf.fill([1, 3], 3.)
print("a: ", a)
print("b: ", b)
print("a + b: ", tf.add(a, b))
print("a - b: ", tf.subtract(a, b))
print("a * b: ", tf.multiply(a, b))
print("b / a: ", tf.divide(b, a))

在这里插入图片描述

原网站

版权声明
本文为[phac123]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_42596275/article/details/126258971