7 - 数学运算之matmul
1个月前 • 155次点击 • 来自 TensorFlow
收录专题: TensorFlow入门笔记
@ matmul的使用
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)
matmul with broadcasting
a = tf.ones([4,2,3])
e = tf.fill([3, 5], 3.0)
g = a @ e
print(g)
标签