![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch.operator
文章平均质量分 68
harry_tea
这个作者很懒,什么都没留下…
展开
-
torch.scatter
本文目录一、函数简介二、二维举例三、详解执行过程1. 第一步2. 第二步3. 第三步4. 问题一、函数简介torch.scatter(input, dim, index, src)dim ([int]) – the axis along which to indexindex (LongTensor) – the indices of elements to scatter, can be either empty or of the same dimensionality as src. Whe原创 2021-10-18 17:21:34 · 6127 阅读 · 0 评论 -
torch.mean
官方文档:pytorch的torch.mean函数有两种用法: 一种是不带任何参数的,他返回的是tensor中所有元素的均值;第二种是带参数的,他返回某一维度的均值,这里分别介绍。原创 2022-09-14 22:59:01 · 2286 阅读 · 0 评论 -
torch.max
官方文档:pytorch的torch.max函数有两种用法: 一种是不带任何参数的,他返回的是tensor中所有元素的最大值;第二种是带参数的,他返回最大值和对应的索引,这里分别介绍。原创 2022-09-14 22:58:30 · 469 阅读 · 0 评论