pytorch深度学习入门—tensor张量的裁剪

Tensor的裁剪可以防止过拟合的出现,也可以有效处理梯度爆炸与梯度消失

torch中可以利用clamp进行梯度裁剪

A.clamp(a,b)表示将A中的元素裁剪到只剩在a—b范围内,原来小于a的元素将赋值为a,大于b的元素将赋值为b

pytorch深度学习入门—tensor张量的裁剪_第1张图片

测试代码:

import torch

a = torch.rand(2, 3) * 10
print(a)
a = a.clamp(5, 8)
print(a)

你可能感兴趣的:(机器学习与深度学习入门,深度学习,pytorch,机器学习)