个人学习笔记
qq_42365882
这个作者很懒,什么都没留下…
展开
-
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 · 61 阅读 · 0 评论 -
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 · 137 阅读 · 0 评论