0, 1로 채워진 텐서를 만드는 방법, 기존 텐서의 크기와 데이터 타입을 그대로 유지하면서 생성
1.ones_like
# 예제 배열 생성
original_array = np.array([[1, 2], [3, 4]])
# np.ones_like를 사용하여 1로 채워진 배열 생성
ones_array = np.ones_like(original_array)
print("Original array:")
print(original_array)
print("\nArray of ones:")
print(ones_array)
# 예제 텐서 생성
original_tensor = torch.tensor([[1, 2], [3, 4]])
# torch.ones_like를 사용하여 1로 채워진 텐서 생성
ones_tensor = torch.ones_like(original_tensor)
print("\nOriginal tensor:")
print(original_tensor)
print("\nTensor of ones:")
print(ones_tensor)
Original array:
[[1 2]
[3 4]]
Array of ones:
[[1 1]
[1 1]]
Original tensor:
tensor([[1, 2],
[3, 4]])
Tensor of ones:
tensor([[1, 1],
[1, 1]])
2.zeros_like
# 예제 배열 생성
original_array = np.array([[1, 2], [3, 4]])
# np.zeros_like를 사용하여 0으로 채워진 배열 생성
zeros_array = np.zeros_like(original_array)
print("Original array:")
print(original_array)
print("\nArray of zeros:")
print(zeros_array)
# 예제 텐서 생성
original_tensor = torch.tensor([[1, 2], [3, 4]])
# torch.zeros_like를 사용하여 0으로 채워진 텐서 생성
zeros_tensor = torch.zeros_like(original_tensor)
print("\nOriginal tensor:")
print(original_tensor)
print("\nTensor of zeros:")
print(zeros_tensor)
Original array:
[[1 2]
[3 4]]
Array of zeros:
[[0 0]
[0 0]]
Original tensor:
tensor([[1, 2],
[3, 4]])
Tensor of zeros:
tensor([[0, 0],
[0, 0]])
300x250
반응형
'프로그래밍 > Python' 카테고리의 다른 글
텐서, 배열 연결하기(numpy, pytorch - concat(concatenate, stack) (1) | 2023.12.04 |
---|---|
텐서 타입 변경하기(numpy - astype(float), pytorch - float()) (1) | 2023.12.04 |
텐서 차원 변경하기(pytorch - squeeze, unsqueeze/ numpy - squeeze, expand_dims) (0) | 2023.12.04 |
배열구조 바꾸기(numpy - reshape, pytorch - view/reshape) (0) | 2023.12.01 |
텐서, 행렬의 곱셈 - numpy(*, dot, @), pytorch(mul, matmul, @) 비교 (0) | 2023.12.01 |