tensors: create tensors vocabulary.
tensors: create file heading
tensors: define tensor constructor.
tensors: add additional constructors.
tensors: add reshaping.
tensors: implement add and include tests.
tensors: add binary operations.
tensors: add scalar multiply.
tensors: added >array functionality
tensors: tests for >array
tensors: unit tests fix
tensors: use more idiomatic >array.
tensors: add multi-methods for scalar multiplication.
tensors: cleaned up >array
tensors: combine a few constructors
tensors: added dims function and unit tests.
tensors: add documentation capabilities.
tensors: added multi-methods for scalar addition/subtraction/division
help.lint.coverage: fix for shadowing "empty" word; prevent the other test-only words from being shadowed too
soundex: move to extra as it's unused; fix authors.txt filename
modify arange to match numpy; replace with naturals
create >float-array for efficient float array construction
use combinators
tensors: documentation added for public functions.
tensors: implement t% and matrix multiplication.
tensors: add slice with non-zero step
tensors: add documentation.
tensors: added transposition funcitonality, with documentation and tests
tensors: add error documentation.
Add error documentation
tensors: fix matmul documentation.
extra/tensors: add tests for arange
tensors: make transpose style more similar
tensors: make some of the PR changes.
tensors: separate shape checking.
tensors: add documentation for non-positive-shape-error.
tensors: add missing comment.
tensors: transpose edits for efficiency
--- /dev/null
+USING: arrays sequences tensors.tensor-slice tools.test ;
+IN: tensors.tensor-slice.tests
+
+{ { 9 7 5 } } [ -1 -6 -2 10 <iota> <step-slice> >array ] unit-test
+{ { 9 7 } } [ -1 -5 -2 10 <iota> <step-slice> >array ] unit-test
+{ { 9 7 } } [ -1 -4 -2 10 <iota> <step-slice> >array ] unit-test
+{ { 9 } } [ -1 -3 -2 10 <iota> <step-slice> >array ] unit-test
+{ { } } [ -4 10 -2 10 <iota> <step-slice> >array ] unit-test
+{ { 6 8 } } [ -4 15 2 10 <iota> <step-slice> >array ] unit-test
+{ { 1 3 } } [ 1 4 2 10 <iota> <step-slice> >array ] unit-test
+{ { 1 3 } } [ 1 5 2 10 <iota> <step-slice> >array ] unit-test
+{ { 1 3 5 } } [ 1 6 2 10 <iota> <step-slice> >array ] unit-test
\ No newline at end of file
--- /dev/null
+USING: accessors kernel locals math math.order sequences ;
+IN: tensors.tensor-slice
+
+TUPLE: step-slice < slice { step integer read-only } ;
+:: <step-slice> ( from to step seq -- step-slice )
+ step zero? [ "can't be zero" throw ] when
+ seq length :> len
+ step 0 > [
+ from [ 0 ] unless*
+ to [ len ] unless*
+ ] [
+ from [ len ] unless*
+ to [ 0 ] unless*
+ ] if
+ [ dup 0 < [ len + ] when 0 len clamp ] bi@
+ ! FIXME: make this work with steps
+ seq dup slice? [ collapse-slice ] when
+ step step-slice boa ;
+
+M: step-slice virtual@
+ [ step>> * ] [ from>> + ] [ seq>> ] tri ;
+
+M: step-slice length
+ [ to>> ] [ from>> - ] [ step>> ] tri
+ dup 0 < [ [ neg 0 max ] dip neg ] when /mod
+ zero? [ 1 + ] unless ;
\ No newline at end of file
--- /dev/null
+! Copyright (C) 2019 HMC Clinic.
+! See http://factorcode.org/license.txt for BSD license.
+USING: arrays help.markup help.syntax math sequences ;
+IN: tensors
+
+ARTICLE: "tensors" "Tensors" "A " { $snippet "tensor" } " is a sequence "
+"of floating point numbers "
+"shaped into an n-dimensional matrix. It supports fast, scalable matrix "
+"operations such as matrix multiplication and transposition as well as a "
+"number of element-wise operations. Words for working with tensors are found "
+"in the " { $vocab-link "tensors" } " vocabulary.\n\n"
+"Tensors can be created "
+"by calling one of four constructors:"
+{ $subsections zeros ones naturals arange }
+"They can be converted to the corresponding N-dimensional array with"
+{ $subsections tensor>array }
+"The number of dimensions can be extracted with:"
+{ $subsections dims }
+"Additionally, tensors can be reshaped with:"
+{ $subsections reshape flatten }
+"Tensors can be combined element-wise with other tensors as well as numbers with:"
+{ $subsections t+ t- t* t/ t% }
+"Finally, tensors support the following matrix operations:"
+{ $subsections matmul transpose } ;
+
+ARTICLE: "tensor-operators" "Tensor Operators" "Info here" ;
+
+HELP: tensor
+{ $class-description "A sequence of floating-point numbers consisting of an "
+{ $snippet "underlying" } " C-style array and a " { $snippet "shape" } "." } ;
+
+HELP: shape-mismatch-error
+{ $values { "shape1" sequence } { "shape2" sequence } }
+{ $description "Throws a " { $link shape-mismatch-error } "." }
+{ $error-description "Thrown by element-wise operations such as " { $link t+ }
+", " { $link t- } ", " { $link t* } ", " { $link t/ } ", and " { $link t% }
+" as well as matrix operations such as " { $link matmul } " if two tensors are "
+"passed and they cannot be combined as desired because of a difference in the "
+"shape." } ;
+
+HELP: non-positive-shape-error
+{ $values { "shape" sequence } }
+{ $description "Throws a " { $link non-positive-shape-error } "." }
+{ $error-description "Thrown by operations such as " { $link zeros } ", "
+{ $link ones } ", " { $link naturals } ", and " { $link reshape }
+", which allow users to directly set the shape of a " { $link tensor }
+", when the shape has zero or negative values." } ;
+
+HELP: zeros
+{ $values { "shape" sequence } { "tensor" tensor } }
+{ $description "Initializes a tensor with shape " { $snippet "shape" }
+" containing all 0s." }
+{ $errors "Throws a " { $link non-positive-shape-error } " if the given "
+"shape has zero or negative values." } ;
+
+HELP: ones
+{ $values { "shape" sequence } { "tensor" tensor } }
+{ $description "Initializes a tensor with shape " { $snippet "shape" }
+" containing all 1s." }
+{ $errors "Throws a " { $link non-positive-shape-error } " if the given "
+"shape has zero or negative values." } ;
+
+HELP: arange
+{ $values { "a" number } { "b" number } { "step" number } { "tensor" tensor } }
+{ $description "Initializes a one-dimensional tensor with values in a range from "
+ { $snippet "a" } " to " { $snippet "b" } " (inclusive) with step-size " { $snippet "step" } "." } ;
+
+HELP: naturals
+{ $values { "shape" sequence } { "tensor" tensor } }
+{ $description "Initializes a tensor with shape " { $snippet "shape" }
+" containing a range of values from 0 to " { $snippet "shape product" } "." }
+{ $errors "Throws a " { $link non-positive-shape-error } " if the given "
+"shape has zero or negative values." } ;
+
+HELP: reshape
+{ $values { "tensor" tensor } { "shape" sequence } { "tensor" tensor } }
+{ $description "Reshapes " { $snippet "tensor" } " to have shape "
+{ $snippet "shape" } "." }
+{ $errors "Throws a " { $link non-positive-shape-error } " if the given "
+"shape has zero or negative values." } ;
+
+HELP: flatten
+{ $values { "tensor" tensor } { "tensor" tensor } }
+{ $description "Reshapes " { $snippet "tensor" } " so that it is one-dimensional." } ;
+
+HELP: dims
+{ $values { "tensor" tensor } { "n" integer } }
+{ $description "Returns the dimension of " { $snippet "tensor" } "." } ;
+
+HELP: t+
+{ $values { "x" { $or tensor number } } { "y" { $or tensor number } } { "tensor" tensor } }
+{ $description "Element-wise addition. Intakes two tensors or a tensor and a number (in either order)." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if passed two tensors that are "
+"not (or cannot be broadcast to be) the same shape." } ;
+
+HELP: t-
+{ $values { "x" { $or tensor number } } { "y" { $or tensor number } } { "tensor" tensor } }
+{ $description "Element-wise subtraction. Intakes two tensors or a tensor and a number (in either order)." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if passed two tensors that are "
+"not (or cannot be broadcast to be) the same shape." } ;
+
+HELP: t*
+{ $values { "x" { $or tensor number } } { "y" { $or tensor number } } { "tensor" tensor } }
+{ $description "Element-wise multiplication. Intakes two tensors or a tensor and a number (in either order)." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if passed two tensors that are "
+"not (or cannot be broadcast to be) the same shape." } ;
+
+HELP: t/
+{ $values { "x" { $or tensor number } } { "y" { $or tensor number } } { "tensor" tensor } }
+{ $description "Element-wise division. Intakes two tensors or a tensor and a number (in either order)." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if passed two tensors that are "
+"not (or cannot be broadcast to be) the same shape." } ;
+
+HELP: t%
+{ $values { "x" { $or tensor number } } { "y" { $or tensor number } } { "tensor" tensor } }
+{ $description "Element-wise modulo operator. Intakes two tensors or a tensor and a number (in either order)." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if passed two tensors that are "
+"not (or cannot be broadcast to be) the same shape." } ;
+
+HELP: tensor>array
+{ $values { "tensor" tensor } { "seq" array } }
+{ $description "Returns " { $snippet "tensor" } " as an n-dimensional array." } ;
+
+HELP: matmul
+{ $values { "tensor1" tensor } { "tensor2" tensor } { "tensor3" tensor } }
+{ $description "Performs n-dimensional matrix multiplication on two tensors, where " { $snippet "tensor1" }
+ " has shape " { $snippet "...xmxn" } " and " { $snippet "tensor1" } " has shape " { $snippet "...xnxp" } "." }
+{ $errors "Throws a " { $link shape-mismatch-error } " if the bottom two "
+"dimensions of the tensors passed do not take the form " { $snippet "mxn" }
+" and " { $snippet "nxp" } " and/or the top dimensions do not match." } ;
+
+HELP: transpose
+{ $values { "tensor" tensor } { "tensor'" tensor } }
+{ $description "Performs n-dimensional matrix transposition on " { $snippet "tens" } "." } ;
+
+ABOUT: "tensors"
--- /dev/null
+! Copyright (C) 2019 HMC Clinic.
+! See http://factorcode.org/license.txt for BSD license.
+USING: accessors alien.c-types kernel math math.order math.vectors
+sequences specialized-arrays tensors tools.test ;
+QUALIFIED-WITH: alien.c-types c
+SPECIALIZED-ARRAY: c:float
+IN: tensors.tests
+
+! Test zeros
+{ float-array{ 0.0 0.0 0.0 0.0 } } [
+ { 4 } zeros vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } zeros shape>>
+] unit-test
+
+{ float-array{ 0.0 0.0 0.0 0.0 } } [
+ { 2 2 } zeros vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } zeros shape>>
+] unit-test
+
+[
+ { 0 5 } zeros
+]
+[ { 0 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+[
+ { -3 5 } zeros
+]
+[ { -3 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+! Test ones
+{ float-array{ 1.0 1.0 1.0 1.0 } } [
+ { 4 } ones vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } ones shape>>
+] unit-test
+
+{ float-array{ 1.0 1.0 1.0 1.0 } } [
+ { 2 2 } ones vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } ones shape>>
+] unit-test
+
+[
+ { 0 5 } ones
+]
+[ { 0 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+[
+ { -3 5 } ones
+]
+[ { -3 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+
+! Test arange
+{ { 4 } float-array{ 0. 1. 2. 3. } } [
+ 0 3 1 arange [ shape>> ] [ vec>> ] bi
+] unit-test
+
+{ { 4 } float-array{ 0. 2. 4. 6. } } [
+ 0 7 2 arange [ shape>> ] [ vec>> ] bi
+] unit-test
+
+{ { 3 } float-array{ 1. 4. 7. } } [
+ 1 9 3 arange [ shape>> ] [ vec>> ] bi
+] unit-test
+
+{ { 5 } float-array{ 1. 3. 5. 7. 9. } } [
+ 1 9 2 arange [ shape>> ] [ vec>> ] bi
+] unit-test
+
+
+! Test naturals
+{ float-array{ 0.0 1.0 2.0 3.0 } } [
+ { 4 } naturals vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 2.0 3.0 } } [
+ { 2 2 } naturals vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } naturals shape>>
+] unit-test
+
+[
+ { 0 5 } naturals
+]
+[ { 0 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+[
+ { -3 5 } naturals
+]
+[ { -3 5 } \ non-positive-shape-error boa = ] must-fail-with
+
+
+! Test reshape
+{ float-array{ 0.0 0.0 0.0 0.0 } } [
+ { 4 } zeros { 2 2 } reshape vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 4 } zeros { 2 2 } reshape shape>>
+] unit-test
+
+[
+ { 2 2 } zeros { 2 3 } reshape
+]
+[ { 2 2 } { 2 3 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 2 2 } zeros { -2 -2 } reshape
+]
+[ { -2 -2 } \ non-positive-shape-error boa = ] must-fail-with
+
+! Test flatten
+{ float-array{ 0.0 0.0 0.0 0.0 } } [
+ { 2 2 } zeros flatten vec>>
+] unit-test
+
+{ { 4 } } [
+ { 2 2 } zeros flatten shape>>
+] unit-test
+
+{ float-array{ 0.0 0.0 0.0 0.0 } } [
+ { 4 } zeros flatten vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } zeros flatten shape>>
+] unit-test
+
+! Test dims
+{ 1 } [
+ { 3 } zeros dims
+] unit-test
+
+{ 2 } [
+ { 2 2 } ones dims
+] unit-test
+
+{ 3 } [
+ { 1 2 3 } zeros dims
+] unit-test
+
+! Test addition
+{ float-array{ 1.0 2.0 3.0 4.0 } } [
+ { 4 } naturals { 4 } ones t+ vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals { 4 } ones t+ shape>>
+] unit-test
+
+{ float-array{ 1.0 2.0 3.0 4.0 } } [
+ { 2 2 } naturals { 2 2 } ones t+ vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } naturals { 2 2 } ones t+ shape>>
+] unit-test
+
+[
+ { 3 } naturals { 2 2 } ones t+ vec>>
+]
+[ { 3 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 4 } naturals { 2 2 } ones t+ vec>>
+]
+[ { 4 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+! Test scalar addition
+{ float-array{ 1.0 2.0 3.0 4.0 } } [
+ { 4 } naturals 1 t+ vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals 1 t+ shape>>
+] unit-test
+
+{ float-array{ 1.0 2.0 3.0 4.0 } } [
+ 1 { 4 } naturals t+ vec>>
+] unit-test
+
+{ { 4 } } [
+ 1 { 4 } naturals t+ shape>>
+] unit-test
+
+! Test subtraction
+{ float-array{ -1.0 0.0 1.0 2.0 } } [
+ { 4 } naturals { 4 } ones t- vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals { 4 } ones t- shape>>
+] unit-test
+
+{ float-array{ -1.0 0.0 1.0 2.0 } } [
+ { 2 2 } naturals { 2 2 } ones t- vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } naturals { 2 2 } ones t- shape>>
+] unit-test
+
+[
+ { 3 } naturals { 2 2 } ones t- vec>>
+]
+[ { 3 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 4 } naturals { 2 2 } ones t- vec>>
+]
+[ { 4 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+! Test scalar subtraction
+{ float-array{ -1.0 0.0 1.0 2.0 } } [
+ { 4 } naturals 1 t- vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals 1 t- shape>>
+] unit-test
+
+{ float-array{ 1.0 0.0 -1.0 -2.0 } } [
+ 1 { 4 } naturals t- vec>>
+] unit-test
+
+{ { 4 } } [
+ 1 { 4 } naturals t- shape>>
+] unit-test
+
+! Test multiplication
+{ float-array{ 0.0 1.0 4.0 9.0 } } [
+ { 4 } naturals { 4 } naturals t* vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals { 4 } naturals t* shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 4.0 9.0 } } [
+ { 2 2 } naturals { 2 2 } naturals t* vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } naturals { 2 2 } naturals t* shape>>
+] unit-test
+
+[
+ { 3 } naturals { 2 2 } naturals t* vec>>
+]
+[ { 3 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 4 } naturals { 2 2 } naturals t* vec>>
+]
+[ { 4 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+! Test division
+{ t } [
+ { 4 } ones
+ { 4 } naturals { 4 } ones t+
+ t/ vec>>
+ { 1.0 0.5 0.33333 0.25 } v-
+ [ abs ] map
+ 0 [ max ] reduce 0.0001 <
+] unit-test
+
+{ { 4 } } [
+ { 4 } ones
+ { 4 } naturals { 4 } ones t+
+ t/ shape>>
+] unit-test
+
+{ t } [
+ { 2 2 } ones
+ { 2 2 } naturals { 2 2 } ones t+
+ t/ vec>>
+ { 1.0 0.5 0.33333 0.25 } v-
+ [ abs ] map
+ 0 [ max ] reduce 0.0001 <
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } ones
+ { 2 2 } naturals { 2 2 } ones t+
+ t/ shape>>
+] unit-test
+
+[
+ { 3 } ones
+ { 2 2 } naturals { 2 2 } ones t+
+ t/ vec>>
+]
+[ { 3 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 4 } ones
+ { 2 2 } naturals { 2 2 } ones t+
+ t/ vec>>
+]
+[ { 4 } { 2 2 } \ shape-mismatch-error boa = ] must-fail-with
+
+! Test scalar division
+{ t } [
+ 1
+ { 4 } naturals { 4 } ones t+
+ t/ vec>>
+ { 1.0 0.5 0.33333 0.25 } v-
+ [ abs ] map
+ 0 [ max ] reduce 0.0001 <
+] unit-test
+
+{ { 4 } } [
+ 1
+ { 4 } naturals { 4 } ones t+
+ t/ shape>>
+] unit-test
+
+{ float-array{ 0.0 0.5 1.0 1.5 } } [
+ { 4 } naturals 2 t/ vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals 2 t/ shape>>
+] unit-test
+
+! Test scalar multiplication
+{ float-array{ 0.0 3.0 6.0 9.0 } } [
+ { 4 } naturals 3 t* vec>>
+] unit-test
+
+{ { 4 } } [
+ { 4 } naturals 3 t* shape>>
+] unit-test
+
+{ float-array{ 0.0 3.0 6.0 9.0 } } [
+ { 2 2 } naturals 3 t* vec>>
+] unit-test
+
+{ { 2 2 } } [
+ { 2 2 } naturals 3 t* shape>>
+] unit-test
+
+{ float-array{ 0.0 3.0 6.0 9.0 } } [
+ 3 { 4 } naturals t* vec>>
+] unit-test
+
+{ { 4 } } [
+ 3 { 4 } naturals t* shape>>
+] unit-test
+
+{ float-array{ 0.0 3.0 6.0 9.0 } } [
+ 3 { 2 2 } naturals t* vec>>
+] unit-test
+
+{ { 2 2 } } [
+ 3 { 2 2 } naturals t* shape>>
+] unit-test
+
+! test mod
+{ float-array{ 0.0 1.0 2.0 0.0 1.0 } } [
+ { 5 } naturals
+ { 5 } ones 3 t*
+ t% vec>>
+] unit-test
+
+{ { 5 } } [
+ { 5 } naturals
+ { 5 } ones 3 t*
+ t% shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 2.0 0.0 1.0 2.0 } } [
+ { 2 3 } naturals
+ { 2 3 } ones 3 t*
+ t% vec>>
+] unit-test
+
+{ { 2 3 } } [
+ { 2 3 } naturals
+ { 2 3 } ones 3 t*
+ t% shape>>
+] unit-test
+
+[
+ { 4 } naturals
+ { 2 3 } ones 3 t*
+ t% vec>>
+]
+[ { 4 } { 2 3 } \ shape-mismatch-error boa = ] must-fail-with
+
+[
+ { 4 } naturals
+ { 2 3 } ones 3 t*
+ t% vec>>
+]
+[ { 4 } { 2 3 } \ shape-mismatch-error boa = ] must-fail-with
+
+! Test scalar mod
+{ float-array{ 0.0 1.0 2.0 0.0 1.0 } } [
+ { 5 } naturals
+ 3
+ t% vec>>
+] unit-test
+
+{ { 5 } } [
+ { 5 } naturals
+ 3
+ t% shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 2.0 0.0 1.0 2.0 } } [
+ { 2 3 } naturals
+ 3
+ t% vec>>
+] unit-test
+
+{ { 2 3 } } [
+ { 2 3 } naturals
+ 3
+ t% shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 0.0 3.0 3.0 } } [
+ 3
+ { 5 } naturals 1 t+
+ t% vec>>
+] unit-test
+
+{ { 5 } } [
+ { 5 } naturals
+ 3
+ t% shape>>
+] unit-test
+
+{ float-array{ 0.0 1.0 0.0 3.0 3.0 3.0 } } [
+ 3
+ { 2 3 } naturals 1 t+
+ t% vec>>
+] unit-test
+
+{ { 2 3 } } [
+ { 2 3 } naturals
+ 3
+ t% shape>>
+] unit-test
+
+! test tensor>array
+{ { 0.0 0.0 } } [
+ { 2 } zeros tensor>array
+] unit-test
+
+{ { { 0.0 0.0 } { 0.0 0.0 } } } [
+ { 2 2 } zeros tensor>array
+] unit-test
+
+{ { { { 1.0 1.0 } { 1.0 1.0 } { 1.0 1.0 } }
+ { { 1.0 1.0 } { 1.0 1.0 } { 1.0 1.0 } } } } [
+ { 2 3 2 } ones tensor>array
+] unit-test
+
+! test matmul
+{ float-array{ 70.0 76.0 82.0 88.0 94.0 190.0 212.0 234.0
+ 256.0 278.0 310.0 348.0 386.0 424.0 462.0 } } [
+ { 3 4 } naturals { 4 5 } naturals matmul vec>>
+] unit-test
+
+{ { 3 5 } } [
+ { 3 4 } naturals { 4 5 } naturals matmul shape>>
+] unit-test
+
+{ float-array{ 70.0 76.0 82.0 88.0 94.0 190.0 212.0 234.0 256.0
+ 278.0 310.0 348.0 386.0 424.0 462.0 1510.0 1564.0
+ 1618.0 1672.0 1726.0 1950.0 2020.0 2090.0 2160.0
+ 2230.0 2390.0 2476.0 2562.0 2648.0 2734.0 } } [
+ { 2 3 4 } naturals { 2 4 5 } naturals matmul vec>>
+] unit-test
+
+{ { 2 3 5 } } [
+ { 2 3 4 } naturals { 2 4 5 } naturals matmul shape>>
+] unit-test
+
+{ float-array{ 70.0 76.0 82.0 88.0 94.0 190.0 212.0 234.0 256.0
+ 278.0 310.0 348.0 386.0 424.0 462.0 1510.0 1564.0 1618.0
+ 1672.0 1726.0 1950.0 2020.0 2090.0 2160.0 2230.0 2390.0 2476.0
+ 2562.0 2648.0 2734.0 4870.0 4972.0 5074.0 5176.0 5278.0 5630.0
+ 5748.0 5866.0 5984.0 6102.0 6390.0 6524.0 6658.0 6792.0 6926.0
+ 10150.0 10300.0 10450.0 10600.0 10750.0 11230.0 11396.0 11562.0
+ 11728.0 11894.0 12310.0 12492.0 12674.0 12856.0 13038.0 } } [
+ { 2 2 3 4 } naturals { 2 2 4 5 } naturals matmul vec>>
+] unit-test
+
+{ { 2 2 3 5 } } [
+ { 2 2 3 4 } naturals { 2 2 4 5 } naturals matmul shape>>
+] unit-test
+
+! test transpose
+{ float-array{ 0.0 2.0 1.0 3.0 } } [
+ { 2 2 } naturals transpose vec>>
+] unit-test
+
+{ float-array{ 0.0 12.0 4.0 16.0 8.0 20.0 1.0
+ 13.0 5.0 17.0 9.0 21.0 2.0 14.0 6.0 18.0
+ 10.0 22.0 3.0 15.0 7.0 19.0 11.0 23.0 } } [
+ { 2 3 4 } naturals transpose vec>>
+] unit-test
+
+{ { 4 3 2 } } [
+ { 2 3 4 } naturals transpose shape>>
+] unit-test
+
+{ t } [
+ { 2 3 4 5 6 } naturals dup transpose transpose =
+] unit-test
--- /dev/null
+! Copyright (C) 2019 HMC Clinic.
+! See http://factorcode.org/license.txt for BSD license.
+USING: accessors alien.c-types alien.data arrays
+concurrency.combinators grouping kernel locals math.functions
+math.ranges math.statistics math multi-methods quotations sequences
+sequences.private specialized-arrays tensors.tensor-slice typed ;
+QUALIFIED-WITH: alien.c-types c
+SPECIALIZED-ARRAY: c:float
+IN: tensors
+
+! Tensor class definition
+TUPLE: tensor
+ { shape array }
+ { vec float-array } ;
+
+! Errors
+ERROR: non-positive-shape-error shape ;
+ERROR: shape-mismatch-error shape1 shape2 ;
+
+<PRIVATE
+
+! Check that the shape has only positive values
+: check-shape ( shape -- shape )
+ dup [ 1 < ] map-find drop [ non-positive-shape-error ] when ;
+
+! Construct a tensor of zeros
+: <tensor> ( shape seq -- tensor )
+ tensor boa ;
+
+: >float-array ( seq -- float-array )
+ c:float >c-array ;
+
+: repetition ( shape const -- tensor )
+ [ check-shape dup product ] dip <repetition>
+ >float-array <tensor> ;
+
+PRIVATE>
+
+! Construct a tensor of zeros
+: zeros ( shape -- tensor )
+ 0 repetition ;
+
+! Construct a tensor of ones
+: ones ( shape -- tensor )
+ 1 repetition ;
+
+! Construct a one-dimensional tensor with values start, start+step,
+! ..., stop (inclusive)
+: arange ( a b step -- tensor )
+ <range> [ length 1array ] keep >float-array <tensor> ;
+
+! Construct a tensors with vec { 0 1 2 ... } and reshape to the desired shape
+: naturals ( shape -- tensor )
+ check-shape [ ] [ product [0,b) >float-array ] bi <tensor> ;
+
+<PRIVATE
+
+: check-reshape ( shape1 shape2 -- shape1 shape2 )
+ 2dup [ product ] bi@ = [ shape-mismatch-error ] unless ;
+
+PRIVATE>
+
+! Reshape the tensor to conform to the new shape
+: reshape ( tensor shape -- tensor )
+ [ dup shape>> ] [ check-shape ] bi* check-reshape nip >>shape ;
+
+! Flatten the tensor so that it is only one-dimensional
+: flatten ( tensor -- tensor )
+ dup shape>>
+ product { } 1sequence >>shape ;
+
+! outputs the number of dimensions of a tensor
+: dims ( tensor -- n )
+ shape>> length ;
+
+! Turn into Factor ND array form
+! Source: shaped-array>array
+TYPED: tensor>array ( tensor: tensor -- seq: array )
+ [ vec>> >array ] [ shape>> ] bi
+ [ rest-slice reverse [ group ] each ] unless-empty ;
+
+<PRIVATE
+
+: check-bop-shape ( shape1 shape2 -- shape )
+ 2dup = [ shape-mismatch-error ] unless drop ;
+
+! Apply the binary operator bop to combine the tensors
+TYPED:: t-bop ( tensor1: tensor tensor2: tensor quot: ( x y -- z ) -- tensor: tensor )
+ tensor1 shape>> tensor2 shape>> check-bop-shape
+ tensor1 vec>> tensor2 vec>> quot 2map <tensor> ; inline
+
+! Apply the operation to the tensor
+TYPED:: t-uop ( tensor: tensor quot: ( x -- y ) -- tensor: tensor )
+ tensor vec>> quot map [ tensor shape>> ] dip <tensor> ; inline
+
+PRIVATE>
+
+! Add a tensor to either another tensor or a scalar
+multi-methods:GENERIC: t+ ( x y -- tensor )
+METHOD: t+ { tensor tensor } [ + ] t-bop ;
+METHOD: t+ { tensor number } [ + ] curry t-uop ;
+METHOD: t+ { number tensor } swap [ + ] curry t-uop ;
+
+! Subtraction between two tensors or a tensor and a scalar
+multi-methods:GENERIC: t- ( x y -- tensor )
+METHOD: t- { tensor tensor } [ - ] t-bop ;
+METHOD: t- { tensor number } [ - ] curry t-uop ;
+METHOD: t- { number tensor } swap [ swap - ] curry t-uop ;
+
+! Multiply a tensor with either another tensor or a scalar
+multi-methods:GENERIC: t* ( x y -- tensor )
+METHOD: t* { tensor tensor } [ * ] t-bop ;
+METHOD: t* { tensor number } [ * ] curry t-uop ;
+METHOD: t* { number tensor } swap [ * ] curry t-uop ;
+
+! Divide two tensors or a tensor and a scalar
+multi-methods:GENERIC: t/ ( x y -- tensor )
+METHOD: t/ { tensor tensor } [ / ] t-bop ;
+METHOD: t/ { tensor number } [ / ] curry t-uop ;
+METHOD: t/ { number tensor } swap [ swap / ] curry t-uop ;
+
+! Divide two tensors or a tensor and a scalar
+multi-methods:GENERIC: t% ( x y -- tensor )
+METHOD: t% { tensor tensor } [ mod ] t-bop ;
+METHOD: t% { tensor number } [ mod ] curry t-uop ;
+METHOD: t% { number tensor } swap [ swap mod ] curry t-uop ;
+
+<PRIVATE
+
+! Check that the tensor has an acceptable shape for matrix multiplication
+: check-matmul-shape ( tensor1 tensor2 -- )
+ [let [ shape>> ] bi@ :> shape2 :> shape1
+ ! Check that the matrices can be multiplied
+ shape1 last shape2 [ length 2 - ] keep nth =
+ ! Check that the other dimensions are equal
+ shape1 2 head* shape2 2 head* = and
+ ! If either is false, raise an error
+ [ shape1 shape2 shape-mismatch-error ] unless ] ;
+
+! Slice out a row from the array
+: row ( arr n i p -- slice )
+ ! Compute the starting index
+ / truncate dupd *
+ ! Compute the ending index
+ swap over +
+ ! Take a slice
+ rot <slice> ;
+
+! Perform matrix multiplication muliplying an
+! mxn matrix with a nxp matrix
+TYPED:: 2d-matmul ( vec1: slice vec2: slice res: slice n: number p: number -- )
+ ! For each element in the range, we want to compute the dot product of the
+ ! corresponding row and column
+ res
+ [ >fixnum
+ ! Get the row
+ [ [ vec1 n ] dip p row ]
+ ! Get the column
+ ! [ p mod vec2 swap p every ] bi
+ [ p mod f p vec2 <step-slice> ] bi
+ ! Take the dot product
+ [ * ] [ + ] 2map-reduce
+ ]
+ map! drop ;
+
+PRIVATE>
+
+
+! Perform matrix multiplication muliplying an
+! ...xmxn matrix with a ...xnxp matrix
+TYPED:: matmul ( tensor1: tensor tensor2: tensor -- tensor3: tensor )
+ ! First check the shape
+ tensor1 tensor2 check-matmul-shape
+
+ ! Now save all of the sizes
+ tensor1 shape>> unclip-last-slice :> n
+ unclip-last-slice :> m :> top-shape
+ tensor2 shape>> last :> p
+ top-shape product :> rest
+
+ ! Now create the new tensor with { 0 ... m*p-1 } repeating
+ top-shape { m p } append naturals m p * t% :> tensor3
+
+ ! Now update the tensor3 to contain the multiplied matricies
+ rest [0,b)
+ [
+ :> i
+ ! First make vec1
+ m n * i * dup m n * + tensor1 vec>> <slice>
+ ! Now make vec2
+ n p * i * dup n p * + tensor2 vec>> <slice>
+ ! Now make the resulting vector
+ m p * i * dup m p * + tensor3 vec>> <slice>
+ ! Push n and p and multiply the clices
+ n p 2d-matmul
+ 0
+ ] map drop
+ tensor3 ;
+
+<PRIVATE
+! helper for transpose: gets the turns a shape into a list of things
+! by which to multiply indices to get a full index
+: ind-mults ( shape -- seq )
+ rest-slice <reversed> cum-product { 1 } prepend ;
+
+! helper for transpose: given shape, flat index, & mults for the shape, gives nd index
+:: trans-index ( ind shape mults -- seq )
+ ! what we use to divide things
+ shape reverse :> S
+ ! accumulator
+ V{ } clone
+ ! loop thru elements & indices of S (mod by elment m)
+ S [| m i |
+ ! we divide by the product of the 1st n elements of S
+ S i head-slice product :> div
+ ! do not mod on the last index
+ i S length 1 - = not :> mod?
+ ! multiply accumulator by mults & sum
+ dup mults [ * ] 2map sum
+ ! subtract from ind & divide
+ ind swap - div /
+ ! mod if necessary
+ mod? [ m mod ] [ ] if
+ ! append to accumulator
+ [ dup ] dip swap push
+ ] each-index
+ reverse ;
+PRIVATE>
+
+! Transpose an n-dimensional tensor
+TYPED:: transpose ( tensor: tensor -- tensor': tensor )
+ ! new shape
+ tensor shape>> reverse :> newshape
+ ! what we multiply by to get indices in the old tensor
+ tensor shape>> ind-mults :> old-mults
+ ! what we multiply to get indices in new tensor
+ newshape ind-mults :> mults
+ ! new tensor of correct shape
+ newshape naturals dup vec>>
+ [ ! go thru each index
+ ! find index in original tensor
+ newshape mults trans-index old-mults [ * ] 2map sum >fixnum
+ ! get that index in original tensor
+ tensor vec>> nth
+ ] map! >>vec ;