wcshds 发表于 2020-4-17 00:57:01

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:27:01

本帖最后由 wcshds 于 2020-6-14 00:28 编辑

之前在官方文档中看到了,@操作符是Python3.5新增的特性,用于矩阵乘法。参见What’s New In Python 3.5。似乎国内的Python教程都很少提及,记录一下。
页: [1]
查看完整版本: Tensorflow中矩阵向量乘法的运算符是如何实现的?