Data.Tensor.Class.ITensor

Overloaded interface for immutable tensors.

Documentation

class Shaped x i => ITensor x i e where

A class for immutable tensors.

Methods

size :: x n e -> Int

Get the numer of elements stored in the tensor.

(//) :: x n e -> [(i, e)] -> x n e

Get a new tensor by replacing the elements at the given indices.

unsafeAt :: x n e -> i -> e

Get the value at the given index, without doing any bounds-checking.

unsafeReplace :: x n e -> [(i, e)] -> x n e

Same as '()' but doesn't do any bounds-checking.

indices :: x n e -> [i]

Get the indices of the elements stored in the tensor.

elems :: x n e -> [e]

Get the elements stored in the tensor.

assocs :: x n e -> [(i, e)]

Get the list of (index, element) pairs stored in the tensor.

tmap :: (e -> e) -> x n e -> x n e

Apply a function elementwise to a tensor.

(*>) :: Num e => e -> x n e -> x n e

Scale every element by the given value.

shift :: Num e => e -> x n e -> x n e

Add a constant to every element.

Instances

BLAS1 e => ITensor Vector Int e

BLAS3 e => ITensor Matrix (Int, Int) e

BLAS3 e => ITensor Banded (Int, Int) e

(!) :: ITensor x i e => x n e -> i -> e

Get the value at the given index. Range-checks the argument.