astrotuna201 / CoreTensor

Dynamic and ranked tensor data structures

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CoreTensor

CoreTensor is a tensor library and a sub-project of the DLVM project.

It provides:

  • Shaping
  • Storage
  • Indexing
  • Slicing
  • Linear algebra shape transformations
  • Broadcasting
  • Collection behavior

Modules

  • CoreTensor implements completely dynamically shaped tensors. Scalars, vectors, matrices and n-D arrays are all represented as Tensor<T>, where T represents the type of each unit. Each Tensor<T> stores a TensorShape, which wraps an array of integers representing the shape of the tensor.

  • RankedTensor implements dynamically shaped but statically ranked tensors. Instead of storing TensorShape (which wraps a dynamically sized array), RankedTensor uses a tuple of integers to represent a shape with known rank. Types T, Tensor1D<T>, Tensor2D<T>, Tensor3D<T>, and Tensor4D<T> represent scalars, vectors, matrices, rank-3 tensors, and rank-4 tensors, respectively.

License

Apache 2.0

About

Dynamic and ranked tensor data structures

License:Apache License 2.0


Languages

Language:Swift 100.0%