I am trying to convert a tensor to numpy array using numpy() function
it is very slow ( takes 50 ms !)
semantic = semantic.cpu().numpy()

semantic is a tensor of size “torch.Size([512, 1024])”  and it’s device is cuda:0