목록einops (1)

Hello Computer Vision

einops rearrange 공부해보기

이번에 ViT를 공부하는 와중에 텐서를 조금 더 내 자유자재로 만들 수 있는 라이브러리를 발견했다 그건 바로 einops! terminal 에 pip install einops 하면은 금방 설치할 수 있다. 한번 코드 예시를 통해 알아보겠다. 매우 직관적이므로 쉽게 알 수 있다. from einops import rearrange import torch.nn as nn import torch img = torch.rand(8, 224, 224, 3) print(img.shape) print(rearrange(img, 'b h w c -> b c h w').shape) print(rearrange(img, 'b h w c -> (b c) h w').shape) print(rearrange(img, 'b h..
딥러닝 2023. 2. 10. 13:41
이전 Prev 1 Next 다음