프로그래밍/Python

0, 1 로 이루어진 배열, 텐서 만들기(pytorch,numpy - ones_like, zeros_like)

히또아빠 2023. 12. 4. 10:30

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
반응형