- 博客(3)
- 收藏
- 关注
原创 tensorflow 学习笔记---- 常量
tf.constant()定义常量。cast()数据类型转换。tf.convert_to_tensor(),标量转换为张量。a为一个维度,shape=(2,)存在两个元素。shape为3行2列,
2023-03-14 14:12:41
78
转载 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
10866
原创 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
3481
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人