Loading

摘要: 假如输入数据是4维的 a = torch.randn(3, 4, 5, 6) y = a[..., 0] # 取第四个维度的第0个 上面操作等同于 y = a[:, :, :, 0] 阅读全文
posted @ 2021-06-11 11:27 Guang'Jun 阅读(62) 评论(0) 推荐(0)