[Pytorch]torch.flatten()

ma-kjh·2024년 8월 13일
0

Pytorch

목록 보기
18/20
post-thumbnail

torch.flatten(input, start_dim=0, end_dim=- 1) → Tensor

torch.flatten 은 입력을 1차원 텐서로 reshape 해준다. start_dimend_dim을 입력해줄 수 있다.

요런식으로 작동한다.

start_dim을 설정해보면

다음과 같은데,

flatten을 시작하는 dimension을 설정해준다.

profile
거인의 어깨에 올라서서 더 넓은 세상을 바라보라 - 아이작 뉴턴

0개의 댓글