该包增加了对CUDA张量类型的支持,实现了与CPU张量相同的功能,但使用GPU进行计算。
它是懒惰的初始化,所以你可以随时导入它,并使用is_available()来确定系统是否支持CUDA。
CUDA语义中有关于使用CUDA的更多细节。
...
AI君
1年前 (2019-03-28) 125℃
0喜欢
此包中包含从Lua Torch移植来的代码。
为了可以使用现有的模型并且方便当前Lua Torch使用者过渡,我们创建了这个包。 可以在torch.legacy.nn中找到nn代码,并在torch.legacy.optim中找到optim代...
AI君
1年前 (2019-03-28) 132℃
0喜欢
封装了multiprocessing模块。用于在相同数据的不同进程中共享视图。
一旦张量或者存储被移动到共享单元(见share_memory_()),它可以不需要任何其他复制操作的发送到其他的进程中。
这个API与原始模型完全兼容,为了让张...
AI君
1年前 (2019-03-28) 171℃
0喜欢
torch.autograd提供了类和函数用来对任意标量函数进行求导。要想使用自动求导,只需要对已有的代码进行微小的改变。只需要将所有的tensor包含进Variable对象中即可。
torch.autograd.backward(vari...
AI君
1年前 (2019-03-28) 97℃
0喜欢
torch.optim是一个实现了各种优化算法的库。大部分常用的方法得到支持,并且接口具备足够的通用性,使得未来能够集成更加复杂的方法。
如何使用optimizer
为了使用torch.optim,你需要构建一个optimizer对象。这个...
AI君
1年前 (2019-03-28) 95℃
0喜欢
torch.nn.init.calculate_gain(nonlinearity,param=None)
对于给定的非线性函数,返回推荐的增益值。这些值如下所示:
nonlinearity
gain
linear
1
...
AI君
1年前 (2019-03-28) 93℃
0喜欢
Convolution 函数
torch.nn.functional.conv1d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
对几个输入平面组...
AI君
1年前 (2019-03-28) 151℃
0喜欢
Parameters
class torch.nn.Parameter()
Variable的一种,常被用于模块参数(module parameter)。
Parameters 是 Variable 的子类。Paramenters和Modu...
AI君
1年前 (2019-03-28) 169℃
0喜欢
一个torch.Storage是一个单一数据类型的连续一维数组。
每个torch.Tensor都有一个对应的、相同数据类型的存储。
class torch.FloatStorage
byte()
将此存储转为byte类型
char()
将...
AI君
1年前 (2019-03-28) 137℃
0喜欢
torch.Tensor是一种包含单一数据类型元素的多维矩阵。
Torch定义了七种CPU tensor类型和八种GPU tensor类型:
Data tyoe
CPU tensor
GPU tensor
32-bit flo...
AI君
1年前 (2019-03-28) 135℃
0喜欢