PyTorch中的torch.max()和torch.maximum()的用法详解
2023-04-18 14:06:07 时间
1、torch.max()
torch.max(input) → Tensor
返回 input tensor 中所有元素的最大值。
torch.max(input, dim, keepdim=False) → output tensors (max, max_indices)
参数:
- input:输入的 tensor。
- dim:按什么维度求最大值(2D中,0代表按列求最大值,1代表按行求最大值)。
- keepdim:是否保持 input tensor 的维度,True 代表 out tensor 与 input tensor 的维度相同,False 代表 out tensor 与 input tensor 的维度不同。
返回值:
- 返回一个 namedtuple (values, indices),values表示指定维度的最大值,indices表示最大值所在的索引,如果给定维度有多个最大值,返回第一个最大值所在的索引。
2、torch.maximum()
torch.maximum(input, other) → Tensor
参数:
- input:第一个输入 tensor。
- other:第二个输入 tensor。
返回值:
- 两个 tensor 进行逐元素比较,返回每个较大的元素组成一个新的 tensor。如果元素中有 NaN 值,则返回 NaN 值。
相关文章
- Python使用tkinter组件Label显示简单数学公式
- 内网渗透之DCOM横向移动
- 以目标为导向的语义交流的共同语言——一个课程学习框架
- python爬虫前奏【成信笔记】
- HTML 5 File API:文件拖放上传功能
- 教你快速创建 Python 虚拟环境
- pyenv 实现Python多版本自由切换
- 用 Python 对 Excel文件进行批量操作
- Python - 接入钉钉机器人
- Python - 抓取 iphone13 pro 线下店供货信息并发送到钉钉机器人,最后设置为定时任务
- crontab - 解决 mac 下通过 crontab 设置了 Python 脚本的定时任务却无法运行
- [源码解析] PyTorch分布式(5) ------ DistributedDataParallel 总述&如何使用
- Python科普系列——类与方法(上篇)
- SAP对STO的交货单执行PGI,报错 -Fld selectn for mvmt type 643 acct 400020 differs
- Spring Boot 实现通用 Auth 认证的 4 种方式
- 盘点4种使用Python批量合并同一文件夹内所有子文件夹下的Excel文件内所有Sheet数据
- OushuDB 学习经验分享(三):技术特点
- Java和Python思维方式的不同之处
- Python中日志记录新技能
- 奥比中光Gemini OpenCV—Python使用