sparse-tensor: typesafe tensor algebra library

[ algebra, data, library, math, mit ] [ Propose Tags ] [ Report a vulnerability ]

This package is intended to be used as a general purpose tensor algebra library. It defines the usual tensor algebra functions such as addition, scalar multiplication, tensor product, and contractions, but also general symmetrizations and further utility functions.

The implemented tensor data type is capable of being used with an arbitrary number of general abstract indices and can incorporate values of any type that allow for a meaningful addition, scaling, and multiplication. The package is thus very flexible and can easily be customised at wish.


[Skip to Readme]

Downloads

Maintainer's Corner

Package maintainers

For package maintainers and hackage trustees

Candidates

Versions [RSS] 0.1.0.0, 0.2, 0.2.1, 0.2.1.1, 0.2.1.2, 0.2.1.3, 0.2.1.4, 0.2.1.5
Change log CHANGELOG.md
Dependencies ad (>=4.2 && <4.5), base (>=4.9 && <5), bytestring (>=0.10 && <0.12), cereal (>=0.4 && <0.6), containers (>=0.5 && <0.7), deepseq (>=1.1 && <1.5), ghc-typelits-knownnat (>=0.2 && <0.8), ghc-typelits-natnormalise (>=0.5 && <0.8), hmatrix (>=0.16.1 && <0.21), parallel (>=3.2 && <3.3), tf-random (>=0.5 && <0.6), zlib (>=0.6 && <0.7) [details]
License MIT
Copyright 2019 Tobias Reinhart and Nils Alex
Author Tobias Reinhart and Nils Alex
Maintainer tobi.reinhart@fau.de, nils.alex@fau.de
Category Data, Math, Algebra
Home page https://github.com/TobiReinhart/sparse-tensor#readme
Source repo head: git clone git://github.com/TobiReinhart/sparse-tensor.git
Uploaded by nalex at 2021-02-27T17:12:35Z
Distributions
Downloads 2871 total (15 in the last 30 days)
Rating (no votes yet) [estimated by Bayesian average]
Your Rating
  • λ
  • λ
  • λ
Status Docs available [build log]
Last success reported on 2021-02-27 [all 1 reports]

Readme for sparse-tensor-0.2.1.5

[back to package description]

The sparse-tensor Package

Hackage-Deps Hackage Build Status

sparse-tensor is a Haskell tensor algebra library. It defines the usual tensor algebra functions such as

  • addition
result = t1 &+ t2
  • scalar multiplication
result = s &. t
  • tensor product
result = t1 &* t2
  • or symmetrizations
result = symTens (0,1) t -- symmetrization in first two indices

The Tensor type

Tensor types can be defined with any value type and index types. For example, a tensor type with n contravariant and m covariant 4-d spacetime indices ranging from 0 to 3 and Rational values can be defined as

type MyTensor n m  = AbsTensor2 n m Ind3 (SField Rational)

The operations on tensors are type-safe, for example it is not possible to add two tensors of different rank,

>>> :set -XDataKinds
>>> (undefined :: MyTensor 0 1) &+ (undefined :: MyTensor 0 2)
<interactive>:3:33: error:
    • Couldn't match type ‘2’ with ‘1’
      [...]

as this causes a type error at compile time.

Predefined tensors

The package comes with pre-defined tensor types. Basic tensors of these types for applications in mathematical physics are exported by Math.Tensor.Examples.Gravity:

>>> sequence_ $ fmap print $ toListT2' delta3  -- print assocs of spacetime delta
(([0],[0]),SField (1 % 1))
(([1],[1]),SField (1 % 1))
(([2],[2]),SField (1 % 1))
(([3],[3]),SField (1 % 1))

>>> sequence_ $ fmap print $ toListT2' eta     -- print assocs of Minkowski metric
(([],[0,0]),SField ((-1) % 1))
(([],[1,1]),SField (1 % 1))
(([],[2,2]),SField (1 % 1))
(([],[3,3]),SField (1 % 1))

>>> let t = invEta &* epsilon
>>> contrATens1 (0,0) $ contrATens1 (1,1) t   -- contraction of inverse eta with epsilon
ZeroTensor

It is of course possible to define further custom tensor types and tensors.

Math.Tensor.LorentzGenerator exports functionality to generate a basis for the space of Lorentz-invariant tensors of certain rank which obey certain symmetries.

Automatic differentiation

sparse-tensor also supports tensors with functions as values. For such tensors, the package also provides the partial function for automatic differentiation. Math.Tensor.Examples.Gravity.Schwarzschild exports the Einstein tensor for a Schwarzschild spacetime, calculated from the Schwarzschild metric:

>>> let e = einstein 2.0 -- Einstein tensor for Schwarzschild metric with r_s = 2.0
>>> e `evalSec` [1.2, 3.1, 1.3, 2.2] -- evaluate at spacetime point
ZeroTensor

Symbolic calculations

The package can also handle symbolic tensor values. All manipulations, including differentiation, are then performed on strings which may be passed to a computer algebra engine. sparse-tensor itself cannot yet simplify these symbolic values. Math.Tensor.Examples.Gravity.SchwarzschildSymbolic exports the Schwarzschild metric with symbolic entries and methods to calculate derived geometric entities:

>>> let g  = schwarzschildS
>>> let g' = schwarzschildS'
>>> let gamma = christoffelS g g'
>>> let comps = toListT2 gamma     -- get assocs
>>> print $ snd $ comps !! 1       -- component gamma^t_tr
SSymbolic "(1 % 2)*((1/(1 - rs/r))*(diff(1 - rs / r, r)))"