JS-Torch - Thư viện JavaScript tương tự PyTorch

• Được phát triển lại từ đầu nhưng vẫn theo cú pháp của PyTorch
• Tích hợp các layer và hàm deep learning, engine vi phân tự động, cùng đối tượng Tensor đầy đủ tính năng có khả năng theo dõi gradient
• Các phép toán Tensor đã được triển khai: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
• Các layer deep learning đã được triển khai: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss

[eduardoleao052]

Cảm ơn bạn đã đăng bài!