Create zero tensor on GPU directly with v0.4/0.5 API

1

As titled, is there any concise/fast way?

2

torch.zeros(sizes, device=‘cuda’)

3

Will that make a difference between this and torch.zeros(sizes).to(device)?

4

torch.zeros(sizes, device=device) won’t copy. functionally they are the same