“Tensor à int python” Réponses codées

Tensor à int python

#in the case the tensor has only one value
int_tensor = torch.IntTensor(3)
int_value = int_tensor.item()
#in the case the tensor has only many values
int_tensor = torch.IntTensor([3,2,1])
list_int_value = int_tensor.tolist()