tf.argmax是一个耗时逐渐累加的操作

最新推荐文章于 2024-04-22 22:20:34 发布
最新推荐文章于 2024-04-22 22:20:34 发布
阅读量712 收藏
点赞数
分类专栏： 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/baidu_26646129/article/details/83657292
版权

在调用训练好的模型进行图片分类时，其中的一个操作是分析logits中数值最大的元素的下标，也就是argmax方法。但在实际运行时发现，随着分类进程的推进，图像分类耗时越来越长，对各部分操作耗时进行统计，发现问题出在tf.argmax()操作。

为了确保分类效率，可以采用numpy库的argmax函数代替之，具体如下：

# 原始的argmax操作
indicies = tf.argmax(logits, 1).eval()

# 修改之后的argmax操作
indicies = np.argmax(logits, axis = 1)
RealMoYe
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
TensorFlow tf.nn.max_pool实现池化操作方式
09-18
本文将详细讨论如何在TensorFlow中使用`tf.nn.max_pool`函数来实现最大值池化操作。 最大值池化（Max Pooling）是一种常见的池化方法，它的主要思想是从输入特征图中选取每个池化窗口内的最大值作为输出。这有助于...
Tensorflow tf.tile()的用法实例分析
09-16
通过以上的分析和实例，我们可以看出`tf.tile()`在TensorFlow中是一个非常实用的工具，它使得在处理张量时能够灵活地复制和扩展数据，而`tf.contrib.seq2seq.tile_batch()`则是针对特定任务（如序列到序列模型的解码...
tf.argmax
weixin_30624825的博客
03-07 58
tf.argmax(input, axis=None, name=None, dimension=None) Returns the index with the largest value across axis of a tensor. input is a Tensor and axis describes which axis of the input Tensor to...
对于tf.argmax的理解
darer49的博客
10-21 274
写一下理解。 1. 举例 现在一个2x3x4的张量a： a = [ [[1, 2, 3, 4], [4, 3, 2, 1], [2, 2, 2, 2]], [[2, 2, 2, 2], [1, 2, 3, 4], [4, 3, 2, 1]] ] 根据a的内容，我们现在假设现在有2个坐标系，每个坐标系有3个坐标轴，每个坐标轴上有4个点，每个点的大小可能是1,2,3,4, 第一个坐标系： x轴点大小分别是1, 2, 3, 4; y轴点大
argmax
最新发布
qq_74693232的博客
04-22 423
argmax是一个在编程中常用的函数，特别是在处理数组或张量（tensor）时。argmax意味着它返回数组中最大值的索引。
tf.argmax()函数
nienelong3319的博客
08-29 501
argmax(input, axis=None, name=None, dimension=None, output_type=tf.int64) Returns the index with the largest value across axes of a tensor. (deprecated arguments) 根据axis的值返回行或者列最大值的下标，axis取值[-2,...
torch.argmax方法详解
wuyalan1994的博客
07-21 4814
torch.argmax方法详解
torch.argmax 的使用
spark的博客
05-12 368
包含四个元素，分别代表第 0 个维度上最大值所在的位置（在本例中，每个元素的值都是 2，表示第 2 行是第一列、第二列、第三列和第四列中的最大值所在的位置）。是一个函数，用于返回张量中某个维度上最大值的索引，即张量中在某个维度上具有最大值的元素所在的位置。在这里，第一个维度的大小为 3，第二个维度的大小为 4。我们在第 0 个维度上使用。例如，如果我们有一个形状为 (3, 4) 的张量。中最大值所在的索引。在 PyTorch 中，
对tf.reduce_sum tensorflow维度上的操作详解
09-20
在TensorFlow库中，`tf.reduce_sum`是一个非常重要的函数，它用于计算张量（tensor）在指定维度上的和。这个函数在处理机器学习模型时，尤其是在计算损失、归一化或者聚合数据时非常常见。下面我们将深入探讨`tf....
tensorflow中tf.slice和tf.gather切片函数的使用
09-18
例如，如果我们有一个三维张量`input`，其形状为`(3, 2, 3)`，那么`tf.slice(input, [1, 0, 0], [1, 1, 3])`将从第二个维度的第一个元素开始，提取一个形状为`(1, 1, 3)`的子张量。 现在，让我们转向`tf.gather`...
【TensorFlow】argmax、equal、accuracy.eval
liudongdong_jlu
08-09 502
1、tf.argmax(vector, 1) 返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。 import tensorflow as tf import numpy as np A = [[1,3,4,5,6]] B = [[1,3,4], [2,4,1]] ...
torch.argmax()
qq_42178122的博客
06-17 207
官方文档： torch.argmax(input, dim, keepdim=False) → LongTensor Returns the indices of the maximum values of a tensor across a dimension. Parameters: input (Tensor) – the input tensor. dim (int) – the dimension to reduce. If None, the argmax of the flattened in
argmax函数_TensorFlow深度学习入门笔记（四）一些基本函数
weixin_39840729的博客
11-25 216
写在前面学习建议：以下学习过程中有不理解可以简单查找下资料，但不必纠结（比如非得深究某一个函数等），尽量快速的学一遍，不求甚解无妨。多实操代码，不能只复制代码，或者感觉懂了就只看。熟能生巧，我亦无他，唯手熟尔今天介绍一些基础函数及其用法，基本全是代码，一些解释都放在代码的注释里了。直接看代码吧，记得在你本地跑一下看哦代码1#tensor.get_shape() 获取tensor的shape，就是维...
机器学习14
lunei
04-17 647
visdom 可视化 import numpy as np import torch import torch.nn.functional as f logits = torch.rand(4, 10) print(logits) pred = f.softmax(logits, dim=1) print(pred) print(pred.sum()) print(pred.shape) pred_label = pred.argmax(dim=1) print('pred_label: '.
tf.argmax的使用
热门推荐
UESTC_C2_403的博客
05-16 7万+
tf.argmax(vector, 1)：返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。 import tensorflow as tf import numpy as np A = [[1,3,4,5,6]] B = [[1,3,4], [2,4,1]] wit
tf.argmax()以及axis解析
萝卜虫的博客
04-23 6万+
用tensorflow做CNN_TEXT文本分类时，看到这个API，然后去官网查了一下，再看了一下别的资料，算是明白它的处理方式了。 首先，明确一点，tf.argmax可以认为就是np.argmax。 　　简单的说，tf.argmax就是返回最大的那个数值所在的下标。 这个很好理解，只是tf.argmax()的参数让人有些迷惑，比如，tf.argmax(array, 1)和tf.argmax(arr
tf.argmax()用法
Jiashilin
04-11 2341
直观理解: tf.argmax(vector, 1)：返回的是vector中的最大值的索引号， 如果vector是一个向量，那就返回一个值， 如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的 import tensorflow as tf import numpy as np A = [[1,3,4,5,6]] B = [[1,3,4], [2,4...
np.argmax()学习
weixin_37721058的博客
11-19 229
[参考链接](https://www.jianshu.com/p/5edeaa3e335b)
深度学习中的tf.nn.softmax(logits, axis=1)以及tf.argmax(prob, axis=1)两个函数的参数以及用法
Childhood_Sweetheart的博客
10-10 1186
参考了下面的两个链接： softmax: https://blog.csdn.net/q2519008/article/details/107086024?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendF
tf.argmax()
05-11
在上面的代码中，`tf.argmax()` 函数的第一个参数是要寻找最大值的张量，第二个参数 `axis=1` 指定了沿着第二个维度（即列）进行计算。因此，`max_indices` 中的第一个元素 3 表示第一行中最大值 4 所在的列索引，第...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值