>>> x = torch.arange(5) >>> x tensor([0, 1, 2, 3, 4]) >>> torch.gt(x,1) # 大于 tensor([0, 0, 1, 1, 1], dtype=torch.uint8) >>> x>1 # 大于 tensor([0, 0, 1, 1, 1], dtype=torch.uint8) >>> torch.ne(x,1) # 不等于 tensor([1, 0, 1, 1, 1], dtype=torch.uint8) >>> x!=1 # 不等于 tensor([1, 0, 1, 1, 1], dtype=torch.uint8) >>> torch.lt(x,3) # 小于 tensor([1, 1, 1, 0, 0], dtype=torch.uint8) >>> x<3 # 小于 tensor([1, 1, 1, 0, 0], dtype=torch.uint8) >>> torch.eq(x,3) # 等于 tensor([0, 0, 0, 1, 0], dtype=torch.uint8) >>> x==3 # 等于 tensor([0, 0, 0, 1, 0], dtype=torch.uint8) 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
Preview:
downloadDownload PNG
downloadDownload JPEG
downloadDownload SVG
Tip: You can change the style, width & colours of the snippet with the inspect tool before clicking Download!
Click to optimize width for Twitter