tensor - Factor Documentation

Vocabulary
tensors

Class description
A sequence of floating-point numbers consisting of an underlying C-style array and a shape.

Definition

USING: arrays specialized-arrays.instances.alien.c-types.float ;

TUPLE: tensor
{ shape array initial: { } }
{ vec float-array initial: float-array{ } } ;

Methods

USING: prettyprint.custom tensors ;

M: tensor >pprint-sequence tensor>array ;

USING: accessors kernel tensors tensors.private ;

M: tensor clone [ shape>> clone ] [ vec>> clone ] bi <tensor> ;

USING: accessors kernel sequences tensors ;

M: tensor clone-like
over tensor?
[ drop clone ] [
[ >tensor ] dip 2dup 2length =
[ shape>> reshape ] [ drop ] if
] if ;

USING: accessors sequences tensors ;

M: tensor length vec>> length ;

USING: accessors arrays kernel sequences
specialized-arrays.instances.alien.c-types.float tensors
tensors.private ;

M: tensor like
over tensor?
[ drop ] [
over float-array?
[ [ dup [ length 1array ] dip <tensor> ] dip ]
[ [ >tensor ] dip ] if 2dup 2length =
[ shape>> reshape ] [ drop ] if
] if ;

USING: accessors arrays kernel sequences tensors ;

M: tensor new-sequence
shape>> 2dup product =
[ nip (tensor) ] [ drop 1array (tensor) ] if ;

USING: accessors sequences tensors tensors.private ;

M: tensor nth num-index vec>> nth ;

USING: accessors sequences.private tensors tensors.private ;

M: tensor nth-unsafe num-index vec>> nth-unsafe ;

USING: prettyprint.custom tensors ;

M: tensor pprint* pprint-object ;

USING: kernel prettyprint.custom tensors ;

M: tensor pprint-delims drop \ t{ \ } ;

USING: accessors sequences tensors tensors.private ;

M: tensor set-nth num-index vec>> set-nth ;

USING: accessors sequences.private tensors tensors.private ;

M: tensor set-nth-unsafe num-index vec>> set-nth-unsafe ;

USING: accessors kernel math sequences tensors tensors.private ;

M: tensor sum
vec>> 0 <simd-slice>
[ simd-slice>> [ sum ] map-sum ] [ end-slice>> sum ] bi + ;