Calculating logarithm of non-zero values in pytorch

[mahsa]

I have a tensor which contains zero and non-zero values. I want replace every non-zero value with its logarithm. Is there any simple way?
Thanks

[ptrblck]

This code should work:

x = torch.arange(0, 10, dtype=torch.float)
idx = x!=0
x[idx] = torch.log(x[idx])

[mahsa]

Thanks @ptrblck