Tensorflow中矩阵向量乘法的运算符是如何实现的?
本帖最后由 wcshds 于 2020-6-14 00:28 编辑import tensorflow as tf
import numpy as np
a = tf.constant(np.arange(6).reshape(2, 3))
b = tf.constant(np.arange(6), shape=(3, 2))
print(tf.matmul(a, b))
print(a@b)
Tensorflow中的矩阵向量乘法既可用tf.matmul()方法也可使用@运算符,请问后者是如何实现的?感觉不太像运算符重载啊。 本帖最后由 wcshds 于 2020-6-14 00:28 编辑
之前在官方文档中看到了,@操作符是Python3.5新增的特性,用于矩阵乘法。参见What’s New In Python 3.5。似乎国内的Python教程都很少提及,记录一下。
页:
[1]