excellent_sun的博客

Pytorch 学习笔记数据操作数据操作

线性回归线性回归的从零开始实现线性回归的简洁实现线性回归的从零开始实现import torchfrom IPython import displayfrom matplotlib import pyplot as pltimport numpy as npimport randomnum_inputs = 2num_examples = 1000w = [2, -3.4]b ...

pytorch 计算flops和paramsfrom thop import profileimport torchfrom resnet_18 import Resnet_18, resnet18model = Resnet_18()input = torch.randn(1, 3, 256, 256)flops, params = profile(model, inputs = ...

tensorboard可视化resnet18可视化resnet18#-*-coding:utf-8-*-import torchimport torchvisionfrom torch.autograd import Variablefrom tensorboardX import SummaryWriter# 模拟输入数据input_data = Variable(torch....

torch.stack用法详解 针对二维数据dim=0,1,2dim=0dim=1dim=2针对三维数据dim=0,1,2,3dim=0或1 是一样的dim=2dim=3针对二维数据dim=0,1,2dim=0dim=1dim=2针对三维数据dim=0,1,2,3dim=0或1 是一样的dim=2dim=3所以可以看出dim是根据数据的维度从最后一维往前取d,d-...