2019年03月的内容

PyTorch手册

PyTorch:torch.cuda

PyTorch:torch.cuda
该包增加了对CUDA张量类型的支持,实现了与CPU张量相同的功能,但使用GPU进行计算。 它是懒惰的初始化,所以你可以随时导入它,并使用is_available()来确定系统是否支持CUDA。 CUDA语义中有关于使用CUDA的更多细节。 ...

AI君 1年前 (2019-03-28) 125℃ 0喜欢

PyTorch手册

PyTorch:遗产包 – torch.legacy

PyTorch:遗产包 – torch.legacy
此包中包含从Lua Torch移植来的代码。 为了可以使用现有的模型并且方便当前Lua Torch使用者过渡,我们创建了这个包。 可以在torch.legacy.nn中找到nn代码,并在torch.legacy.optim中找到optim代...

AI君 1年前 (2019-03-28) 132℃ 0喜欢

PyTorch手册

PyTorch:torch.multiprocessing

PyTorch:torch.multiprocessing
封装了multiprocessing模块。用于在相同数据的不同进程中共享视图。 一旦张量或者存储被移动到共享单元(见share_memory_()),它可以不需要任何其他复制操作的发送到其他的进程中。 这个API与原始模型完全兼容,为了让张...

AI君 1年前 (2019-03-28) 171℃ 0喜欢

PyTorch手册

PyTorch:torch.optim

PyTorch:torch.optim
torch.optim是一个实现了各种优化算法的库。大部分常用的方法得到支持,并且接口具备足够的通用性,使得未来能够集成更加复杂的方法。 如何使用optimizer 为了使用torch.optim,你需要构建一个optimizer对象。这个...

AI君 1年前 (2019-03-28) 95℃ 0喜欢

PyTorch手册

PyTorch:torch.nn.init

PyTorch:torch.nn.init
torch.nn.init.calculate_gain(nonlinearity,param=None) 对于给定的非线性函数,返回推荐的增益值。这些值如下所示: nonlinearity gain linear 1 ...

AI君 1年前 (2019-03-28) 93℃ 0喜欢

PyTorch手册

PyTorch:torch.nn

PyTorch:torch.nn
Parameters class torch.nn.Parameter() Variable的一种,常被用于模块参数(module parameter)。 Parameters 是 Variable 的子类。Paramenters和Modu...

AI君 1年前 (2019-03-28) 169℃ 0喜欢

PyTorch手册

PyTorch:torch.Storage

PyTorch:torch.Storage
一个torch.Storage是一个单一数据类型的连续一维数组。 每个torch.Tensor都有一个对应的、相同数据类型的存储。 class torch.FloatStorage byte() 将此存储转为byte类型 char() 将...

AI君 1年前 (2019-03-28) 137℃ 0喜欢

PyTorch手册

PyTorch:torch.Tensor

PyTorch:torch.Tensor
torch.Tensor是一种包含单一数据类型元素的多维矩阵。 Torch定义了七种CPU tensor类型和八种GPU tensor类型: Data tyoe CPU tensor GPU tensor 32-bit flo...

AI君 1年前 (2019-03-28) 135℃ 0喜欢