Skip to content

Lightweight package for sharing tensor-algebra definitions

License

Notifications You must be signed in to change notification settings

JuliaMath/TensorCore.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

46 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TensorCore.jl

Stable CI Codecov

This package is intended as a lightweight foundation for tensor operations across the Julia ecosystem. Currently it exports three operations:

  • hadamard elementwise multiplication, with unicode operator ,
  • tensor product preserves all dimensions, operator , and
  • boxdot contracts neighbouring dimensions, named after the unicode .
julia> using TensorCore

julia> A = [1 2 3; 4 5 6]; B = [7 8 9; 0 10 20];

julia> A  B  # hadamard(A, B)
2×3 Matrix{Int64}:
 7  16   27
 0  50  120

julia> V = [1, 10];

julia> C = A  V  # tensor(A, V)
2×3×2 Array{Int64, 3}:
[:, :, 1] =
 1  2  3
 4  5  6

[:, :, 2] =
 10  20  30
 40  50  60

julia> summary(A  B)
"2×3×2×3 Array{Int64, 4}"

julia> C  V  # boxdot(C, V)
2×3 Matrix{Int64}:
 101  202  303
 404  505  606

julia> summary(C  rand(2,5,7))
"2×3×5×7 Array{Float64, 4}"