Tensorflow.js tf.any()函数
Tensorflow.js 是谷歌开发的一个开源库,用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还帮助开发人员用 JavaScript 语言开发 ML 模型,以便 ML 可以直接在浏览器或 Node.js 中使用。
tf.any()函数主要用于计算 tf.Tensor 给定维度上元素的逻辑或。
语法:
tf.all(x, axis, keepDims)
参数:
- x :要输入的张量。
- Axis :要减少的维度。
- keepDims :如果为真,则保留大小为 1 的缩减元素。
返回值:返回 tf.Tensor
示例 1:
Javascript
// Importing @tensorflow/tfslibrary
const tf = require("@tensorflow/tfjs")
// Initialization of tensor1d
const gfg=tf.tensor1d([-1, -2, -3, -4], 'bool')
// Printing the boolean value
gfg.any().print();
Javascript
// Importing @tensorflow/tfslibrary
const tf = require("@tensorflow/tfjs")
// Initialization of the axis
const axis = 1;
// Initialization 2d tensor of shape [2,2]
const gfg=tf.tensor2d([1, 1, 0, 0], [2, 2], 'bool')
gfg.any(axis).print()
输出:
Tensor
true
示例 2:
Javascript
// Importing @tensorflow/tfslibrary
const tf = require("@tensorflow/tfjs")
// Initialization of the axis
const axis = 1;
// Initialization 2d tensor of shape [2,2]
const gfg=tf.tensor2d([1, 1, 0, 0], [2, 2], 'bool')
gfg.any(axis).print()
输出:
Tensor
[True, False]
参考: https://js.tensorflow.org/api/latest/#any