Python
learning!!!!
给我一瓶AC钙
是一颗萌菜没错了
展开
-
numpy数据类型dtype转换
numpy中的数据类型转换,使用函数astype(),不能直接改原数据的dtypeb = np.array([1, 2, 3, 4])print(b.dtype)b = b.astype('float64')print(b)print(b.dtype)'''int32[1. 2. 3. 4.]float64'''原创 2021-02-06 21:00:58 · 374 阅读 · 0 评论 -
Pytorch 由多个不同的网络模型组合而成的大模型(用一个损失函数,和优化方法去训练)
一开始写模型的时候,网上学习到的都是那种非常简单的模型,可是我要实现的模型是将多个模型套起来,一个大模型,一起训练。其实只要先定义几个小模型,然后将其定义在大模型的init中,然后用forward来控制内部的运作即可,直接看代码把。class Model_one(nn.Module): def __init__(self, n_input, n_output): super(Model_one, self).__init__() self.predict = nn原创 2020-11-30 19:50:54 · 18362 阅读 · 14 评论 -
Pytroch 对矩阵形状的变化和拼接
主要用到view和cat两个函数,这里就简单地使用1.torch.view()主要是用来改变矩阵的形态,比如现在是一个[5,3]的矩阵,那我们就可以使用view函数将[5,3]的矩阵变成[3,5] , [1,15]或等等的形态out = torch.tensor([[0.1],[0.2],[0.3]])print(out.size())out = out.view(-1, len(out))print(out.size())print(out) Print:torch.Siz原创 2020-11-30 18:43:23 · 630 阅读 · 0 评论