JS-Torch - Library Javascript yang mirip dengan PyTorch
(github.com/eduardoleao052)- Dikembangkan ulang dari nol sambil mengikuti sintaks PyTorch
- Menyertakan layer dan fungsi deep learning, mesin diferensiasi otomatis, serta objek Tensor berfitur lengkap yang mendukung pelacakan gradient
- Operasi Tensor yang telah diimplementasikan: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
- Layer deep learning yang telah diimplementasikan: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss
1 komentar
Terima kasih sudah membagikan ini!