9P by xguru 2달전 | favorite | 댓글 1개
  • PyTorch 의 문법을 따르면서 처음부터 새로 개발
  • 딥러닝 Layers와 함수, 자동 미분(Differentiation) 엔진, Gradient 추적이 가능한 완전한 기능의 Tensor 객체를 내장
  • 구현된 Tensor Operation: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
  • 구현된 딥러닝 Layers: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss