摘要: torch.max(input) → Tensor 返回输入tensor中所有元素的最大值 a = torch.randn(1, 3)>>0.4729 -0.2266 -0.2085 torch.max(a)>>0.4729 torch.max(input, dim, keepdim=False,  阅读全文
posted @ 2020-04-25 21:58 地球上最后一个直男 阅读(1225) 评论(0) 推荐(0) 编辑