- 博客(2)
- 收藏
- 关注
RSS订阅原创 2021-09-26
_, predict = torch.max(b, dim=1) 定义一个数组 b = torch.tensor([[1, 2, 3, 4, 5, 6], [5, 6, 7, 8, 9, 0], [2, 3, 4, 5, 6, 7], [3, 4, 1, 2, 7, 6]]) _, predict = torch.max(b, dim=0)
2021-09-26 14:46:02
78
原创 2021-09-25
python中的数组切分 数组切分，a[:, [-1]]，a[:-1, :-1] #先定义一个数组 a = np.array([[1, 2, 3, 4, 5, 6], [5, 6, 7, 8, 9, 0], [2, 3, 4, 5, 6, 7], [3, 4, 1, 2, 7, 6]]) #1:3表示取第2，3两行, 2:4表示取第2，3，4三列 print(a[1:3, 1:4]) ''' result: [[6 7
2021-09-25 15:20:32
151
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人