Your browser does not seem to support JavaScript. As a result, your viewing experience will be diminished, and you have been placed in read-only mode.
Please download a browser that supports JavaScript, or enable it if it's disabled (i.e. NoScript).
pytorch常用函数备忘录
本教程记录一些pytorch经常使用的函数, 这些函数在编写深度学习模型时非常有用.
torch.clamp
这个函数的作用就是把矩阵的所有元素, 限定在某个区间之内:
import torch a = torch.Tensor([2, 1, 0.5, -0.5]) a = torch.clamp(a, 0, 1)
此时a就被限定在了0-1区间.
与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待