7 - 数学运算之matmul

10个月前 • 416次点击 • 来自 算法

收录专题: TensorFlow入门笔记

标签: TensorFlow

a = tf.ones([4, 2, 3])
b = tf.fill([4, 3, 5], 2.)
c = a @ b
print(c)
d = tf.matmul(a, b)
print(d)

a = tf.ones([4,2,3])
e = tf.fill([3, 5], 3.0)
g = a @ e
print(g)