class torch.utils.data.Dataset
表示Dataset的抽象类。
所有其他数据集都应该进行子类化。所有子类应该override__len__和__getitem__,前者提供了数据集的大小,后者支持整数索引,范围...
AI君
1年前 (2019-03-28) 117℃
0喜欢
torch.utils.model_zoo.load_url(url, model_dir=None)
在给定URL上加载Torch序列化对象。
如果对象已经存在于 model_dir 中,则将被反序列化并返回。URL的文件名部分应遵循命...
AI君
1年前 (2019-03-28) 128℃
0喜欢
该包增加了对CUDA张量类型的支持,实现了与CPU张量相同的功能,但使用GPU进行计算。
它是懒惰的初始化,所以你可以随时导入它,并使用is_available()来确定系统是否支持CUDA。
CUDA语义中有关于使用CUDA的更多细节。
...
AI君
1年前 (2019-03-28) 125℃
0喜欢
torch.utils.ffi.create_extension(name, headers, sources, verbose=True, with_cuda=False, package=False, relative_to='.', ...
AI君
1年前 (2019-03-28) 116℃
0喜欢
此包中包含从Lua Torch移植来的代码。
为了可以使用现有的模型并且方便当前Lua Torch使用者过渡,我们创建了这个包。 可以在torch.legacy.nn中找到nn代码,并在torch.legacy.optim中找到optim代...
AI君
1年前 (2019-03-28) 132℃
0喜欢
torch.autograd提供了类和函数用来对任意标量函数进行求导。要想使用自动求导,只需要对已有的代码进行微小的改变。只需要将所有的tensor包含进Variable对象中即可。
torch.autograd.backward(vari...
AI君
1年前 (2019-03-28) 97℃
0喜欢
封装了multiprocessing模块。用于在相同数据的不同进程中共享视图。
一旦张量或者存储被移动到共享单元(见share_memory_()),它可以不需要任何其他复制操作的发送到其他的进程中。
这个API与原始模型完全兼容,为了让张...
AI君
1年前 (2019-03-28) 171℃
0喜欢
torch.optim是一个实现了各种优化算法的库。大部分常用的方法得到支持,并且接口具备足够的通用性,使得未来能够集成更加复杂的方法。
如何使用optimizer
为了使用torch.optim,你需要构建一个optimizer对象。这个...
AI君
1年前 (2019-03-28) 95℃
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喜欢
torch.nn.init.calculate_gain(nonlinearity,param=None)
对于给定的非线性函数,返回推荐的增益值。这些值如下所示:
nonlinearity
gain
linear
1
...
AI君
1年前 (2019-03-28) 93℃
0喜欢