Just found interesting thing torch.argmax(0,0,0) will return 2 instead logically 0 (torch.max also). So be aware of this when you will be using it...
確定! 回上一頁