pytorch常用函数备忘录

---

本教程记录一些pytorch经常使用的函数, 这些函数在编写深度学习模型时非常有用.

1. torch.clamp:

这个函数的作用就是把矩阵的所有元素, 限定在某个区间之内:

import torch
a = torch.Tensor([2, 1, 0.5, -0.5])
a = torch.clamp(a, 0, 1)

此时a就被限定在了0-1区间.