JS-Torch - ไลบรารี Javascript ที่คล้ายกับ PyTorch
(github.com/eduardoleao052)- พัฒนาขึ้นใหม่ทั้งหมดตั้งแต่ต้น โดยยึดตามไวยากรณ์ของ PyTorch
- มาพร้อมอ็อบเจ็กต์ Tensor ที่มีความสามารถครบถ้วน รองรับเลเยอร์และฟังก์ชันสำหรับดีปเลิร์นนิง, เอนจินการดิฟเฟอเรนชิเอชันอัตโนมัติ และการติดตาม Gradient
- Tensor Operation ที่รองรับ: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
- เลเยอร์ดีปเลิร์นนิงที่รองรับ: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss
1 ความคิดเห็น
ขอบคุณที่โพสต์!