当前位置: 首页 > news >正文

PyTorch:.max(1)和.max(0)的使用

目录

1).max(1)的使用:

2).max(0)的使用:


1).max(1)的使用:

假设有一个形状为 ( m , n ) 的 Tensor x ,其中m表示行数,n表示列数。

 x.max(1) ,相当于x.max(dim=1) 。作用:沿着第 1 维(即列数)的方向,返回 x 的每行最大值和对应的列索引。

import torch# 构造一个 3x3 的 Tensor x
x = torch.tensor([[0.2, 0.9, 0.1], [0.8, 0.4, 0.3], [0.2, 0.7, 0.6]])###### .max(1)的用法
values1, indices1 = x.max(1)
print(values1)   # 每一行的最大值
print(indices1)   # 每一行最大值所对应的列索引####### 以下是输出结果:
# tensor([0.9000, 0.8000, 0.7000])
# tensor([1, 0, 1])

2).max(0)的使用:

假设有一个形状为 ( m , n ) 的 Tensor x ,其中m表示行数,n表示列数。

 x.max(0) ,相当于x.max(dim=0) 。作用:沿着第 0 维(即行数)的方向,返回 x 的每列最大值和对应的行索引。

import torch# 构造一个 3x3 的Tensor x
x = torch.tensor([[0.2, 0.9, 0.1], [0.8, 0.4, 0.3], [0.2, 0.7, 0.6]])###### .max(0)的用法
values2, indices2 = x.max(0)
print(values2)   # 每一列的最大值
print(indices2)   # 每一列最大值所对应的行索引####### 以下是输出结果:
# tensor([0.8000, 0.9000, 0.6000])
# tensor([1, 0, 2])

http://www.lryc.cn/news/518998.html

相关文章:

  • ASP.NET Core 中使用 Cookie 身份验证
  • Ollama私有化部署大语言模型LLM
  • 安卓app抓包总结(精)
  • Three.js 性能优化:打造流畅高效的3D应用
  • PHP 在 2025 年的现状与展望
  • 力扣经典二分题:4. 寻找两个正序数组的中位数
  • 解决WordPress出现Fatal error: Uncaught TypeError: ftp_nlist()致命问题
  • Excel 技巧07 - 如何计算到两个日期之间的工作日数?(★)如何排除节假日计算两个日期之间的工作日数?
  • 快速实现一个快递物流管理系统:实时更新与状态追踪
  • kvm 解决 安装windows 虚拟机cpu 核数问题
  • Ansys Fluent Aeroacoustics 应用
  • 119.使用AI Agent解决问题:Jenkins build Pipeline时,提示npm ERR! errno FETCH_ERROR
  • istio-proxy内存指标
  • List详解 - 双向链表的操作
  • 多目标优化算法之一:基于分解的方法
  • conntrack iptables 安全组
  • stringRedisTemplate.execute执行lua脚本
  • HDFS异构存储和存储策略
  • 生成idea ui风格界面代码
  • 嵌入式C语言:二维数组
  • 【机器学习:四、多输入变量的回归问题】
  • JVM实战—OOM的定位和解决
  • iOS 本地新项目上传git仓库,并使用sourceTree管理
  • mysql之基本select语句 运算符 排序分页
  • 如何在 Ubuntu 22.04 上安装 Nagios 服务器教程
  • 数据库事务:确保数据一致性的关键机制
  • 词作词汇积累:错付、大而无当、语焉不详、愈演愈烈
  • selenium学习笔记
  • asp.net core webapi 并发请求时 怎么保证实时获取的用户信息是此次请求的?
  • 实时数仓:基于数据湖的实时数仓与数据治理架构