Torch.jl:在Julia中公开暴露火炬的明智扩展

时间:2024-05-21 03:40:45
【文件属性】:

文件名称:Torch.jl:在Julia中公开暴露火炬的明智扩展

文件大小:70KB

文件格式:ZIP

更新时间:2024-05-21 03:40:45

C++

火炬 揭露Julia中的火炬的明智扩展。 该软件包旨在提供Tensor类型,该类型将所有计算到用C ++编写的的基础张量库ATen上。 笔记: 需要一台具有CUDA GPU(CUDA 10.1或更高版本)的计算机在没有GPU的情况下将需要惰性工件功能 快速开始 要添加软件包,请从Julia REPL中输入] ,输入Pkg提示符,然后执行以下操作: pkg > add Torch 或通过Julia的包裹经理Pkg。 julia > using Pkg; Pkg . add ( " Torch " ); 使用范例 using Metalhead, Metalhead . Flux, Torch using Torch : torch resnet = ResNet () 我们可以通过简单的torch调用将对象移至Torch tresnet = resnet . layers |


【文件预览】:
Torch.jl-master
----Project.toml(1KB)
----Manifest.toml(3KB)
----.github()
--------workflows()
----test()
--------test_nnlib.jl(7KB)
--------runtests.jl(642B)
----LICENSE(5KB)
----.gitlab-ci.yml(391B)
----bors.toml(28B)
----src()
--------statistics.jl(724B)
--------Torch.jl(1KB)
--------grads.jl(5KB)
--------normalise.jl(2KB)
--------tensor.jl(5KB)
--------nnlib.jl(2KB)
--------wrap()
--------error.jl(340B)
--------scalar.jl(229B)
--------ops.jl(8KB)
--------broadcast.jl(2KB)
--------utils.jl(66B)
----README.md(2KB)
----build()
--------torch_api_generated.cpp.h(275KB)
--------torch_api.h(4KB)
--------CMakeLists.txt(366B)
--------torch_api.cpp(17KB)
--------README.md(821B)
--------torch_api_generated.h(90KB)

网友评论