DeepLearning
Rain_Alley
这个作者很懒,什么都没留下…
展开
-
tensorflow 学习笔记---- 常量
tf.constant()定义常量。cast()数据类型转换。tf.convert_to_tensor(),标量转换为张量。a为一个维度,shape=(2,)存在两个元素。shape为3行2列,原创 2023-03-14 14:12:41 · 85 阅读 · 0 评论 -
PyTorch 保存和读取训练好的模型
#保存torch.save(the_model.state_dict(), PATH)#读取the_model = TheModelClass(*args, **kwargs)the_model.load_state_dict(torch.load(PATH))原创 2019-05-11 11:58:55 · 3522 阅读 · 2 评论 -
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)pr...转载 2019-05-12 10:26:57 · 10893 阅读 · 0 评论