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

Tensorflow.js 是一个开源库，用于在 Javascript 中创建机器学习模型，允许用户直接在浏览器中运行模型。

tf.fill() 是在类 tf.Tensor 中定义的函数。它用于创建一个填充有标量值的张量。

用法:

tf.fill( shape, value, dtype )

参数：

shape:它是一个定义输出张量形状的整数数组。
value:它是一个标量值，用于填充输出张量。
dtype:它定义了输出张量中元素的数据类型。它可以是 ‘float32’|'int32'|'bool'|'complex64'|'string'。包含是可选的，默认值为 ‘float32’。

返回值：它返回填充有标量值的指定形状的张量。

示例 1：用标量数填充张量

创建一个形状为 [4, 2] 的张量，填充为标量值 2。
它将输出张量中元素的默认数据类型作为浮点数。

Javascript


// Dynamic loading the "@tensorflow/tfjs" module
const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
  
// Creating a tensor of of shape [4,2] filled with 
// scalar value 2
var matrix = tf.fill(shape = [4,2],value = 2)
    
// Printing the tensor
matrix.print()

输出：

Tensor
    [[2, 2],
     [2, 2],
     [2, 2],
     [2, 2]]

示例 2：显式定义元素的数据类型

创建一个形状为 [3, 4] 的张量，其中填充了字符串“Gfg”。

Javascript


// Dynamic loading the "@tensorflow/tfjs" module
const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
  
// Creating a tensor  of shape [3,4] filled
// with string value 'Gfg'
var matrix = tf.fill(shape = [3, 4], 
        value = 'Gfg', dtype = 'string')
    
// Printing the tensor
matrix.print()

输出：

Tensor
    [['Gfg', 'Gfg', 'Gfg', 'Gfg'],
     ['Gfg', 'Gfg', 'Gfg', 'Gfg'],
     ['Gfg', 'Gfg', 'Gfg', 'Gfg']]

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

相关用法

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