“Tensor do int Pythona” Kod odpowiedzi

Tensor do int Pythona

#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()