logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

(6) cuda中的矩阵相乘

1的时候MatmulKernel函数每次会计算一个位置的结果。2的时候MatmulKernel函数每次会计算4个位置的结果。无论block dim多大,每个线程都只计算一个位置的结果。矩阵乘法是啥,在这里就不多说了。当我们的block dim是1。当block dim是2。

文章图片
#矩阵#线性代数#c++ +1
Pytorch 引入albumentations对ImageFolder读取的数据集进行数据增强

前言albumentations作为第三方数据增强库。在我们使用ImageFolder加载数据集的时候不能直接使用。我们对ImageFolder进行改造,使其可以使用albumentations进行数据增强。代码如下(示例):from torchvision.datasets.folder import *from matplotlib import pyplot as pltimport cv2

#pytorch#深度学习
到底了