>>> x = torch.randn(3, 2) >>> y = torch.ones(3, 2) >>> torch.where(x > 0, x, y) tensor([[1.4013, 1.0000], [1.0000, 0.9267], [1.0000, 0.4302]]) >>> x tensor([[ 1.4013, -0.9960], [-0.3715, 0.9267], [-0.7163, 0.4302]]) 1 2 3 4 5 6 7 8 9 10
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