
pytorch
文章平均质量分 51
pytorch相关
HLBayes
QQ:446425957
展开
-
pytorch指定使用GPU
默认使用0号GPU。如果需要使用其他GPU,需要人为指定。原创 2024-04-10 20:16:22 · 253 阅读 · 0 评论 -
torch 卷积、反卷积、pooling输出大小
卷积Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1)Hnew=⌊Hold+2×padding−dilation×(kernel−1)−1stride+1⌋=⌊Hold+2×0−1×(kernel−1)−11+1⌋\begin{aligned}\textcolor{blue}{H_{new}}&=\left\lfloor \frac{\textcolor{blue}{H_{old}}原创 2022-04-10 17:06:04 · 961 阅读 · 0 评论 -
torch.rot90 的理解与使用
torch.rot90 的理解与使用原创 2022-09-22 11:16:52 · 2169 阅读 · 1 评论