- 博客(3)
- 收藏
- 关注
原创 pytorch学习记录1-torch.max及torch.unsqueeze()的用法
1. torch.max(x,0):返回一列里的最大值及对应的索引。2. torch.max(x,1):返回一行里的最大值及对应的索引。3. torch.unsqueeze(input,dim):指定一个dim的位置插入一个维度为1的tensor
2022-04-11 19:06:04
966
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人