pytorch
明月清风あ
这个作者很懒,什么都没留下…
展开
-
pytorch中的gather函数
先来看官方文档的解释:Gathers values along an axis specified by dim.For a 3-D tensor the output is specified by:out[i][j][k] = input[index[i][j][k]][j][k] # if dim == 0out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1out[i][j][k] = input[i][j][index原创 2020-10-20 20:08:57 · 152 阅读 · 0 评论 -
二维互相关运算实现图像中的物体边缘检测
在二维卷积层中,一个二维输入数组和一个二维核(kernel)数组通过互相关运算输出一个二维数组。如下图所示,输入是一个高和宽均为3的二维数组。我们将该数组的形状记为3×3或(3,3)。核数组的高和宽分别为2。该数组在卷积计算中又称卷积核或过滤器(filter)。卷积核窗口(又称卷积窗口)的形状取决于卷积核的高和宽,即2×2。在二维互相关运算中,卷积窗口从输入数组的最左上方开始,按从左往右、从上往下的顺序,依次在输入数组上滑动。当卷积窗口滑动到某一位置时,窗口中的输入子数组与核数组按元素相乘并求和,得原创 2020-10-18 10:43:31 · 940 阅读 · 0 评论 -
kaggle预测房价
import torchimport torch.nn as nnimport numpy as npimport pandas as pdimport syssys.path.append("..")import d2lzh_pytorch as d2l# 获取和读取数据集torch.set_default_tensor_type(torch.FloatTensor)train_data = pd.read_csv('D:/predictHousePrice/train.csv').原创 2020-10-16 21:37:54 · 157 阅读 · 0 评论 -
pytorch中的Datasets及DataLoader
Pytorch中有工具函数torch.utils.Data.DataLoader,通过这个函数我们在准备加载数据集使用mini-batch的时候可以使用多线程并行处理,这样可以加快我们准备数据集的速度。Datasets就是构建这个工具函数的实例参数之一。from torch.utils.data import Dataset, DataLoader如果想要继承Datasets,父类中的两个私有成员函数必须被重载。def getitem(self, index):def len(self):转载 2020-10-15 20:46:16 · 626 阅读 · 0 评论