在学习Pytorch的时候,有些函数的功能还是不知道,书里也不可能全部详尽的介绍,因此,需要自己多查多看。
- Pytorch:torch.ge()、torch.gt()、torch.le()、torch.lt() 函数的作用
- https://blog.csdn.net/weixin_40522801/article/details/106579068
- item()函数是用来获取tensor的元素
- .t() 是 .transpose函数的简写版,区别在于只能对二维进行转置
torch.no_grad()
是一个上下文管理器,被该语句 wrap 起来的部分将不会track 梯度。-
assert 声明,主张 ,作用:检查程序,不符合声明的,即终止程序