Notice
Recent Posts
Recent Comments
Link
Tags
- CP
- argmax
- math
- maths
- Axis
- SUAPC
- 패스트캠퍼스 #포트폴리오 #직장인자기계발 #환급챌린지 #포트폴리오챌린지 #패스트캠퍼스후기 #초격차패키지 #오공완
- probability
- laplace
- sinchon icpc
- Dim
- Discrete
- randn
- #패스트캠퍼스 #패캠챌린지 #수강료0원챌린지 #패캠챌린지 #직장인인강 #직장인자기계발 #패캠인강후기 #패스트캠퍼스후기 #환급챌린지 #본인이선택한강의명
- dims
- Computer science
- probability theory
- Counting
- cs-theory
- pytorch
Archives
목록Dim (1)
Piico의 일상
PyTorch confusables 대잔치 ep.1 [pytorch dim]
What are dims in pytorch?- "dim" stands for dimensions- same as numpy "axis"- for a 2D torch.Tensor, dim==0 goes along the rows, dim==1 goes along the colsCode:a = torch.randn(5, 5)b = torch.argmax(a) # torch.argmax(input, dim=None, keepdim=False)Output:tensor([[-8.4741e-01, 3.6448e-01, 1.4155e+00, 2.6417e-01, -3.4608e-01], [-8.9281e-01, -3.1258e-01, 1.4784e+00, 2.9023e-01, -1.0445e+..
딥러닝 (Deep Learning)
2024. 5. 9. 15:03