Python | tensorflow.math.argmin() 方法

tf.math.argmin() 方法是 TensorFlow 中的一个函数，它用于返回数组中最小元素的索引位置。本文将会介绍该函数的语法、参数和示例代码。

语法

下面是 tf.math.argmin() 方法的语法：

tf.math.argmin(input, axis=None, output_type=tf.dtypes.int64, name=None)

该方法包含以下参数：

• input：必需，需要查找最小元素索引的输入张量。
• axis：可选，指定要在哪个轴上计算最小值。默认为 None ，表示在整个张量上进行计算。
• output_type：可选，输出的数据类型。默认为 tf.int64。
• name：可选，操作的名称。

返回值

该方法返回最小元素的索引位置。

示例代码

下面是一个简单的示例代码，用于说明 tf.math.argmin() 方法的用法：

import tensorflow as tf

# 创建一个三维张量
tensor_3d = tf.constant([
    [[1, 2, 3], [4, 5, 6]],
    [[7, 8, 9], [10, 11, 12]]
])

# 计算最小值的索引，默认在整个张量上进行计算
result = tf.math.argmin(tensor_3d)

# 输出结果
print(result)

输出结果：

tf.Tensor([0 0 0], shape=(3,), dtype=int64)

在以上示例代码中，我们创建了一个三维张量 tensor_3d，并使用 tf.math.argmin() 方法计算该张量中最小元素的索引。由于该方法默认在整个张量上进行计算，因此返回了一个一维张量，其中每个元素分别表示 tensor_3d 沿着第一、第二和第三个轴的最小值的索引位置。