[python] 파이썬 PyTorch를 사용하여 텐서를 생성하는 방법은?
- 텐서를 생성하기 위해
torch
모듈을 임포트합니다.import torch
- 리스트, 튜플 또는 배열을 사용하여 데이터를 포함한 텐서를 생성할 수 있습니다.
data = [1, 2, 3, 4, 5] tensor = torch.tensor(data)
- 크기를 지정하여 빈 텐서를 생성할 수도 있습니다. 이때 데이터는 초기화되지 않습니다.
size = (3, 3) empty_tensor = torch.empty(size)
- 모든 요소가 0으로 초기화된 텐서를 생성하기 위해
zeros
함수를 사용할 수 있습니다.zeros_tensor = torch.zeros(size)
- 모든 요소가 1로 초기화된 텐서를 생성하기 위해
ones
함수를 사용할 수 있습니다.ones_tensor = torch.ones(size)
- 특정 값으로 초기화된 텐서를 생성하기 위해
full
함수를 사용할 수 있습니다.value = 5 full_tensor = torch.full(size, value)
- 특정 범위 내에서 일정한 간격으로 값을 가진 텐서를 생성하기 위해
arange
함수를 사용할 수 있습니다.start = 0 end = 10 step = 2 range_tensor = torch.arange(start, end, step)
- 지정된 범위 내에서 무작위로 값을 가지는 텐서를 생성하기 위해
rand
함수를 사용할 수 있습니다.random_tensor = torch.rand(size)
이렇게 파이썬 PyTorch를 사용하여 텐서를 생성할 수 있습니다. 텐서는 신경망 모델 등 다양한 머신러닝 작업에서 사용되며 데이터를 저장하고 처리하기 위한 중요한 자료구조입니다. 더 자세한 내용은 PyTorch 공식 문서를 참조하시기 바랍니다.