Skip to content

Instantly share code, notes, and snippets.

@mratsim
Created May 5, 2018 15:40
Show Gist options
  • Save mratsim/e83f16b5ff5060032a73de99812db91c to your computer and use it in GitHub Desktop.
Save mratsim/e83f16b5ff5060032a73de99812db91c to your computer and use it in GitHub Desktop.

Core tensors operations support

-------------------------------------------------- --------- --------------------------- --------------------------
Feature Tensor CudaTensor ClTensor
-------------------------------------------------- --------- --------------------------- --------------------------
Accessing tensor properties [x] [x] [x] Tensor creation [x] by converting a cpu Tensor by converting a cpu Tensor Accessing or modifying a single value [x] [] [] Iterating on a Tensor [x] [] [] Slicing a Tensor [x] [x] [x] Slice mutation a[1,_] = 10 [x] [] [] Comparison == [x] [] [] Element-wise basic operations [x] [x] [x] Universal functions [x] [] [] Automatically broadcasted operations [x] [x] [x] Matrix-Matrix and Matrix vector multiplication [x] [x] [x] Displaying a tensor [x] [x] [x] Higher-order functions (map, apply, reduce, fold) [x] internal only internal only Transposing [x] [x] [] Converting to contiguous [x] [x] [] Reshaping [x] [x] [] Explicit broadcast [x] [x] [x] Permuting dimensions [x] [] [] Concatenating along existing dimensions [x] [] [] Squeezing singleton dimensions [x] [x] [] Slicing + squeezing in one operation [x] [] []

-------------------------------------------------- --------- --------------------------- --------------------------

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment