diff --git a/cn-Book/4.从零开始实现一个用于文本生成的 GPT 模型.md b/cn-Book/4.从零开始实现一个用于文本生成的 GPT 模型.md index c1f3987..d1d4bab 100644 --- a/cn-Book/4.从零开始实现一个用于文本生成的 GPT 模型.md +++ b/cn-Book/4.从零开始实现一个用于文本生成的 GPT 模型.md @@ -159,7 +159,7 @@ print(batch) ```python tensor([[ 6109, 3626, 6100, 345], #A - [ 6109, 1110, 6622, 257]]) + [ 6109, 1110, 6622, 257]]) #A 第一行对应第一段文本,第二行对应第二段文本。 ``` @@ -263,11 +263,11 @@ print("Variance:\n", var) ```python Mean: - tensor([[0.1324], + tensor([[0.1324], [0.2170]], grad_fn=) Variance: - tensor([[0.0231], + tensor([[0.0231], [0.0398]], grad_fn=) ```