텐서 타입 변경하기(numpy - astype(float), pytorch - float())

# NumPy 배열 생성
numpy_array = np.array([1, 2, 3])

# NumPy 배열을 다른 데이터 타입으로 변환
float_array = numpy_array.astype(float)
int_array = numpy_array.astype(int)

print("Original NumPy array:", numpy_array)
print("Float array:", float_array)
print("Integer array:", int_array)

# PyTorch 텐서 생성
torch_tensor = torch.tensor([1, 2, 3])

# PyTorch 텐서를 다른 데이터 타입으로 변환
float_tensor = torch_tensor.float()
int_tensor = torch_tensor.int()

print("Original PyTorch tensor:", torch_tensor)
print("Float tensor:", float_tensor)
print("Integer tensor:", int_tensor)

Original NumPy array: [1 2 3]
Float array: [1. 2. 3.]
Integer array: [1 2 3]
Original PyTorch tensor: tensor([1, 2, 3])
Float tensor: tensor([1., 2., 3.])
Integer tensor: tensor([1, 2, 3], dtype=torch.int32)