PyTorch碎片
文章平均质量分 81
柚有所思
Paperreading, think and research
展开
-
PyTorch碎片:PyToch和Torchvision对应版本
前言错误分析:安装pytorch或torchvision时,无法找到对应版本cuda可以找到,但是无法转为.cuda()以上两种或类似错误,一般由两个原因可供分析:cuda版本不合适,重新安装cuda和cudnnpytorch和torchvision版本没对应上pytorch和torchvision版本对应关系pytorchtorchvisionpythoncuda1.5.10.6.1>=3.69.2, 10.1,10.21.5.0原创 2020-06-21 11:48:04 · 69420 阅读 · 7 评论 -
PyTorch碎片:深刻透彻理解Torch中Tensor.contiguous()函数
1.函数定义Returns a contiguous tensor containing the same data as self tensor.返回一个与原始tensor相同元素数据的 “连续”tensor类型If self tensor is contiguous, this function returns the self tensor.如果原始tensor本身就是连续的,则返...原创 2020-04-12 18:55:04 · 2507 阅读 · 1 评论 -
PyTorch碎片:F.pad的图文透彻理解
1. F.pad函数定义F.pad是pytorch内置的tensor扩充函数,便于对数据集图像或中间层特征进行维度扩充,下面是pytorch官方给出的函数定义。torch.nn.functional.pad(input, pad, mode='constant', value=0)函数变量说明:input需要扩充的tensor,可以是图像数据,抑或是特征矩阵数据pad扩充维度,用...原创 2020-04-03 18:38:54 · 25115 阅读 · 22 评论