torch.ones, zeros, full, empty에 대하여

dusruddl2·2022년 9월 22일
0

pytorch tutorial

목록 보기
2/4

torch.ones(size,...)
: 1값을 갖는 텐서

torch.zeros(size,...)
: 0값을 갖는 텐서

torch.zeros(size, fill_value, ...)
: fill_value값을 갖는 텐서

torch.empty(size,...)
: 쓰레기 값을 갖는 텐서

a = torch.ones(2,3)
b = torch.zeros(2,3)
c = torch.full((2,3),2)
d = torch.empty(2,3) # 텐서 성분의 값들은 쓰레기값

print(a)
print(b)
print(c)
print(d)


torch.zeros_like(input,...)
: 입력 텐서와 동일한 모양의 텐서를 만들어주는데, 0으로 채워줘

torch.ones_like(input,...)
: 입력 텐서와 동일한 모양의 텐서를 만들어주는데, 1으로 채워줘

torch.full_like(input,fill_value,...)
: 입력 텐서와 동일한 모양의 텐서를 만들어주는데, fill_value으로 채워줘

torch.empty_like(input,...)
: 입력 텐서와 동일한 모양의 텐서를 만들어주는데, 쓰레기 값으로 채워줘

e = torch.zeros_like(c)
f = torch.ones_like(c)
g = torch.full_like(c,3)
h = torch.empty_like(c)
print(e)
print(f)
print(g)
print(h)

profile
정리된 글은 https://dusruddl2.tistory.com/로 이동

0개의 댓글