
基础函数
AD稳稳
这个作者很懒,什么都没留下…
展开
-
Pytorch基础函数(三)基本数学函数(1)——max()与min()函数(对dim参数的详解)
这篇博客介绍两个基本的数学函数,max()与min()两个。从函数名就可以看出这两个函数的作用:max():查找最大值min():查找最小值笔者一开始遇到这两个函数,感觉这两个简单函数对Tensor(张量)的操作还是有一些弯弯绕。我们来梳理一下,梳理出来就会清楚很多。一、max()函数函数定义:torch.max(input, dim, max=None, max_indice...原创 2020-04-02 23:09:42 · 9042 阅读 · 2 评论 -
Pytorch基础函数(二)变形函数——view()与view_as()
view()函数是用于对Tensor(张量)进行形状变化的函数,如一个Tensor的size是3x2,可以通过view()函数将其形状转为2x3。但是需要注意的是进行操作的张量必须是contiguous()的,即在在内存中连续的。(不连续可以使用tensor.contiguous()操作转为连续)。一、view()函数基本操作函数定义:view(*args) [Ps:...原创 2020-04-02 18:25:42 · 12867 阅读 · 1 评论 -
Pytorch基础函数(一)生成随机张量——rand()与randn()
在PyTorch中,基本的数据单元为Tensor(张量),而对于Tensor的随机初始化,常用的就是rand()与randn()两个函数。一、rand()函数torch.rand(*size,out=None)原创 2020-04-02 10:35:50 · 16253 阅读 · 0 评论