Tensorflow.js tf.matMul()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf.matMul()函数用于计算两个矩阵A * B的点积。

用法:

tf.matMul (a, b, transposeA?, transposeB?)

参数：此函数接受如下所示的参数：

a:这是点积运算中的第一个矩阵。
b:这是点积运算中的第二个矩阵。
transposeA:这是可选的，如果将其设置为true，则在乘法之前先对a进行转置。
transposeB:这是可选的，如果将其设置为true，则b在乘法之前会转置。

返回值：它返回两个矩阵的点积。

下面的示例说明了tf.matMul()函数的用法。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor of some elements
let geek1 = tf.tensor2d([2, 1], [1, 2]);
let geek2 = tf.tensor2d([11, 12, 13, 14], [2, 2]);
  
// Calling the .avgPool3d() function over
// the above tensor as its parameter and 
// printing the result.
geek1.matMul(geek2).print();

输出：

Tensor
     [[35, 38],]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor of some elements
let geek1 = tf.tensor2d([2, 1], [1, 2]);
let geek2 = tf.tensor2d([61, 62, 63, 64], [2, 2]);
  
// Calling the .avgPool3d() function over
// the above tensor as its parameter and 
// printing the result.
tf.matMul(geek1, geek2).print();

输出：

Tensor
     [[185, 188],]

参考：https://js.tensorflow.org/api/latest/#matMul

相关用法

注：本文由纯净天空筛选整理自thacker_shahid大神的英文原创作品 Tensorflow.js tf.matMul() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。