torch.full() 함수 – 주어진 크기의 텐서를 특정 값으로 채워줌

`torch.full()` 함수는 주어진 크기(size)의 새로운 텐서를 생성하고 모든 요소를 지정된 값(fill_value)으로 채웁니다.

함수 시그니처는 다음과 같습니다:

torch.full(size, fill_value, dtype=None, layout=torch.strided, device=None, requires_grad=False, pin_memory=False) -> Tensor

– `size`: 텐서의 크기를 나타내는 정수 또는 튜플입니다.
– `fill_value`: 텐서의 모든 요소에 할당할 값입니다.
– `dtype` (선택적): 텐서의 데이터 유형을 나타내는 torch.dtype 객체입니다. 기본값은 None이며, 이 경우에는 입력값(fill_value)의 데이터 유형이 사용됩니다.
– `layout` (선택적): 텐서의 레이아웃을 나타내는 torch.layout 열거형입니다. 기본값은 `torch.strided`입니다.
– `device` (선택적): 텐서가 생성될 장치를 지정하는 torch.device 객체입니다. 기본값은 None이며, 이 경우에는 현재 기본 장치가 사용됩니다.
– `requires_grad` (선택적): 텐서가 기울기(gradient) 계산을 지원해야 하는지 여부를 나타내는 불리언 값입니다. 기본값은 False입니다.
– `pin_memory` (선택적): 텐서를 핀 메모리(pinned memory)에 할당해야 하는지 여부를 나타내는 불리언 값입니다. 기본값은 False입니다.

`torch.full()` 함수를 사용하면 주어진 크기의 텐서를 특정 값으로 채우는 것이 간단하고 편리해집니다. 예를 들어, `torch.full((3, 4), 5)`는 3×4 크기의 텐서를 생성하고 모든 요소를 5로 채웁니다.

댓글 남기기

이메일은 공개되지 않습니다. 필수 입력창은 * 로 표시되어 있습니다