torch

译者:cluster

torch package 包含 多维张量和之前定义过数学运算的数据结构。此外,它提供了支持许多高效序列化的张量(或者任意类型) 的程序集,以及一些有用工具。

它支持CUDA环境,使你可以在NVIDIA GPU进行张量运算(要求compute capability 版本3.0及以上)