- 博客(6)
- 收藏
- 关注
RSS订阅原创 pytorch 查看tensor的数据类型
import torchx=torch.Tensor([1,2])print('x: ',x)print('type(x): ',type(x))print('x.dtype: ',x.dtype) # x的具体类型y=x.int()print('y: ',y)print('type(y): ',type(y))print('y.dtype: ',y.d...
2019-02-22 14:58:15
36617
原创 torch.nn.Linear()函数的理解
import torchx = torch.randn(128, 20) # 输入的维度是（128，20）m = torch.nn.Linear(20, 30) # 20,30是指维度output = m(x)print('m.weight.shape:\n ', m.weight.shape)print('m.bias.shape:\n', m.bias.shape)print...
2019-02-21 15:10:19
146956
19
原创 torch.nn 和torch.nn.functional的区别浅析
举例：拿maxpool操作来说import torchx_input = torch.randn(20, 16, 50)# 先定义一个model，再把数据传进去# pool of size=3, stride=2m = torch.nn.MaxPool1d(3, stride=2)x_output = m(x_input)print(x_output.shape) # torc...
2019-02-21 10:47:51
1238
原创 torch.nn.Conv2d()函数详解
import torchx = torch.randn(2,1,7,3)conv = torch.nn.conv2d(1,8,(2,3))res = conv(x)print(res.shape) # shape = (2, 8, 6, 1)输入：x[ batch_size, channels, height_1, width_1 ]batch_size ...
2019-02-21 09:52:02
56076
14
原创 logging模块，记录函数运行时间
import loggingimport timedef log(func): def wrapper(*args, **kwatgs): start_time = time.time() logging.info('start %s()' % func.__name__) ret = func(*args, **kwatgs) ...
2019-02-16 09:42:53
1950
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人