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

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

.broadcastTo() 函数用于将数组循环到 NumPy-style 的一致模型。

注意：

• 这里，张量的形状等同于广播从最后到开始的形状。其中，1 是张量形状的前缀，只要它具有与广播形状相同的长度即可。
• 如果 input.shape[i]==shape[i]，那么第 (i+1) 轴以前与广播一致，如果 input.shape[i]==1 加上 shape[i] ==N，那么指定的输入张量被该轴覆盖 N 次。

用法：

tf.broadcastTo(x, shape)

Parameters: 

• x:它是指定的张量输入，可以是 tf.Tensor、TypedArray 或 Array 类型。
• shape:它是输入要广播到的指定形状。

返回值：它返回tf.Tensor对象。

范例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const y = tf.tensor1d([1, 2, 3, 4]);
  
// Calling broadcastTo() method and
// Printing output
tf.broadcastTo(y, [1, 4]).print();

输出：

Tensor
     [[1, 2, 3, 4],]

范例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling broadcastTo() method and
// Printing output
tf.broadcastTo(tf.tensor1d([3.6, 5.8, 3.7, 1.4, 9.3, 10.5]), 
                           [1, 6]).print();

输出：

Tensor
     [[3.5999999, 5.8000002, 3.7, 1.4, 9.3000002, 10.5],]

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