Answers for "tensor to int python"

0

tensor to 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()
Posted by: Guest on July-05-2021

Python Answers by Framework

Browse Popular Code Answers by Language