PyTorch中部分方法介绍

1.torchvision.transforms.Normalize(mean, std)

mean参数:给定序列的均值,形式为(R,G,B)(并不一定是三维)

std参数:给定序列的平均标准偏差(标准差),形式为(R,G,B)(并不一定是三维)

功能:将给定的Tensor正则化,也就是按照如下公式计算:

\frac{Tensor-mean}{std}


2.torch.utils.data.DataLoader()


3.torch.nn.ReLu(flag)

   torch.nn.Tanh()


4.a.data.cpu()和a.cpu().data一样,都是将数据取出放到cpu,准备放到cpu计算


5.torch.clamp(input, min, max, out=None),将input的tensor限定在min~max之间,如果小于这个范围,取min;大于这个范围,去max,亦可input.clamp(min,max)这样使用


6.nn.MSELoss(reduce),使用是传入两个量x,y,计算(x-y)^2,如果reduce=False,直接返回向量形式的 loss;如果reduce=True,返回标量

7.nn.MaxPool2d(2)中,默认步长是2

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值