1. Prism.js 代码语法高亮
以下是一个 Python 机器学习示例:
import torch
import torch.nn as nn
class TransformerBlock(nn.Module):
"""Multi-head self-attention + FFN"""
def __init__(self, d_model=512, n_heads=8, d_ff=2048):
super().__init__()
self.attention = nn.MultiheadAttention(d_model, n_heads)
self.