下载 Torch

Torch是一个基于Lua语言的科学计算框架，提供了丰富的算法和工具库，广泛应用于深度学习领域。

安装

要使用Torch，需要先安装LuaJIT和LuaRocks。可以使用以下命令在Mac或Linux系统上安装：

curl -s https://raw.githubusercontent.com/torch/ezinstall/master/install-deps | bash

然后，可以使用以下命令安装Torch：

git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install-deps
./install.sh

使用

安装完成后，可以使用以下命令启动Torch交互式环境：

th

也可以将Torch集成到Python中进行开发：

import torch

示例

以下是一个使用Torch进行简单线性回归的示例：

-- 生成数据集
x = torch.linspace(0, 1, 100)
y = x * 2 + torch.randn(100) * 0.2

-- 定义模型和损失函数
model = torch.nn.Linear(1, 1)
criterion = torch.nn.MSECriterion()

-- 定义优化器
sgd = torch.optim.SGD(model.parameters, lr=0.01)

-- 训练模型
for i=1,1000 do
    -- 将数据传入模型
    output = model:forward(x:view(-1, 1))

    -- 计算损失函数
    loss = criterion:forward(output, y:view(-1, 1))

    -- 反向传播
    gradOutput = criterion:backward(output, y:view(-1, 1))
    model:backward(x:view(-1, 1), gradOutput)

    -- 更新权重
    sgd.step(sgd)
end

-- 测试模型
test_x = torch.Tensor{{0.5}}
test_y = model:forward(test_x)
print(test_y[1])