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

Tensorflow.js 是 Google 开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。 Tensorflow.js tf.layers.dot() 函数用于在提供的两个张量之间应用点积。

用法：

tf.layers.dot(args);

参数：该函数接受以下参数：

args:它是具有以下字段的对象：
- axes:它定义了点积的坐标轴。
- normalize:它是用于以余弦值形式生成点积答案的布尔数据。
- inputShape:它定义了模型输入层的形状。它用于创建输入层。
- dtype:它是图层的数据类型。它用于模型的第一层。
- name:字符串是输入层的名称。
- weight:它是张量，其值为层的初始值。
- trainable:它定义了层是否可训练。它是一种布尔数据类型。
- batchInputShape:它为输入层中的样本定义了批次的形状。它用于制作输入层。
- batchSize:在输入层的构建中作为batchInputShape的补充。它用于制作输入层。
- inputDType:它是层中输入数据的数据类型。

返回值：它返回两个张量的乘积张量。

以下是此函数的一些示例。

范例1：在这个例子中，我们将简单地制作带有 -2 轴旋转值的张量的点积。

Javascript


// import * as tf from "@tensorflow/tfjs"
  
// Generating tensor of [3,3] shape and size.
const geek_tens1 = tf.randomUniform([3 ,3]);
const geek_tens2 = tf.randomUniform([3 ,3]);
  
// Making tensor campatiable to apply to dot product.
const geek_Input = [geek_tens1 ,geek_tens2];
  
// Calling dot product
const geek_config = {axes:-2};
const geek_DotPro = tf.layers.dot(geek_config);
const geek_DotProRes = geek_DotPro.apply(geek_Input);
  
// Printing our result
console.log(geek_DotProRes);
geek_DotProRes.print();

输出：

Tensor
    [[0.7034817],
     [0.2338114],
     [1.1493738]]
Tensor
    [[0.7034817],
     [0.2338114],
     [1.1493738]]

范例2：在这个例子中，我们将制作两个张量的点积，并以余弦形式查看结果。

Javascript


import * as tf from "@tensorflow/tfjs"
  
// Collect both outputs and print separately.
// Generating tensor of [3,3] shape and size.
const geek_tens1 = tf.randomUniform([2 ,4]);
const geek_tens2 = tf.randomUniform([2 ,4]);
  
// Making tensor campatiable to apply to dot product.
const geek_Input = [geek_tens1 ,geek_tens2];
  
// Calling dot product
const geek_config = {axis:1, normalize:true,
        dtype:'int32', name:'DotProduct' };
const geek_DotPro = tf.layers.dot(geek_config);
const geek_DotProRes = geek_DotPro.apply(geek_Input);
  
// Printing our result
console.log(geek_DotProRes);

输出：

Tensor
    0.6817111968994141

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

相关用法

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