序列化
将对象保存到磁盘文件。 | |
从文件中加载用torch.save()保存的对象 |
线程数
返回用于并行化CPU操作的线程数 | |
设置CPU上用于操作内并行的线程数。 | |
返回CPU上用于操作间并行的线程数。 | |
设置用于互操作并行性的线程数。 |
禁用和启用梯度计算
上下文管理器有助于禁用和启用梯度计算
上下文管理器禁用梯度计算. | |
上下文管理器启用梯度计算. | |
上下文管理器将“梯度计算”设置为“开”或“关”。 | |
如果当前启用梯度模式,则返回True。 | |
上下文管理器启用或禁用推理模式 | |
如果当前启用了推理模式,则返回True。 |
>>> x = torch.zeros(1, requires_grad=True)
>>> with torch.no_grad():
... y = x * 2
>>> y.requires_grad
False
>>> is_train = False
>>> with torch.set_grad_enabled(is_train):
... y = x * 2
>>> y.requires_grad
False
>>> torch.set_grad_enabled(True) # this can also be used as a function
>>> y = x * 2
>>> y.requires_grad
True
>>> torch.set_grad_enabled(False)
>>> y = x * 2
>>> y.requires_grad
False