torch
Elvira521yan
这个作者很懒,什么都没留下…
展开
-
torch与torchvision的版本选择与对应
历史版本最新版本linux pip安装(其他参考官网, 见上)# ROCM 4.2 (Linux only)pip install torch==1.9.0+rocm4.2 torchvision==0.10.0+rocm4.2 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html# ROCM 4.1 (Linux only)pip install torch==1.9.0+rocm4.1 tor原创 2021-12-21 10:31:59 · 4832 阅读 · 0 评论 -
torch.pow
torch.pow(input, exponent, *, out=None) → Tensor计算两个张量或者一个张量与一个标量的指数计算结果,返回一个张量。input和exponent都可以是张量或者标量,1)若input和exponent都为张量,则必须维度一致;2)若input和exponent其中一个为标量,一个为张量,标量以广播的形式进行计算例子:>>> a = torch.randn(4)>>> atensor([ 0.原创 2021-07-06 21:30:57 · 10438 阅读 · 0 评论 -
torch.arange
torch.arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor返回一个一维向量,其大小为,取值区间为[start,end),从start开始,以step为步长增加,直到end结束(不包括end)例子:>>> torch.arange(5)tensor([ 0, 1, 2...原创 2021-07-06 21:14:52 · 5368 阅读 · 0 评论 -
torch.utils.data.DataLoader
DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, *, prefetch_factor=2, persi...原创 2021-07-06 17:35:40 · 218 阅读 · 0 评论 -
xavier_normal_和xavier_normal的区别
新旧名字的差别:新的-xavier_normal_ 旧的-xavier_normalfrom torch.nn.init import xavier_normal_, xavier_normaltorch/nn/init.pyxavier_normal = _make_deprecate(xavier_normal_)_make_deprecate:对于使用 old_name 的情况给出 warnings :建议使用后面加下划线的方法,将old_name 函数重定向到 new.原创 2021-07-06 10:11:41 · 3068 阅读 · 0 评论 -
torch.nn.init
https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-nn/#torchnn原创 2021-07-05 23:19:57 · 209 阅读 · 0 评论 -
Torch.empty
torch.empty(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False, pin_memory=False, memory_format=torch.contiguous_format) → Tensor返回:一个填满未初始化数据的张量(张量的形状是由变量参数size定义的)参数:size (int...) - 整数序列,其定义了输出张量的形状;可以是一个可..翻译 2021-07-05 23:14:16 · 2602 阅读 · 0 评论