|
|
a9c113248a
|
Take references as input for Tensor::cat.
|
2023-06-25 13:03:05 +01:00 |
|
|
|
5e03a1bc29
|
One more test.
|
2023-06-25 10:57:46 +01:00 |
|
|
|
ba0693a908
|
Fix the reduce_sum implementation and add some tests.
|
2023-06-25 10:55:04 +01:00 |
|
|
|
7ccf27dda2
|
More general broadcast setup.
|
2023-06-25 08:55:09 +01:00 |
|
|
|
6b2cd9c51c
|
Add the broadcast operator.
|
2023-06-24 19:16:03 +01:00 |
|
|
|
fbbf3951dd
|
More narrow testing.
|
2023-06-24 15:10:31 +01:00 |
|
|
|
0f34738831
|
Fix the cpu implementation for narrow.
|
2023-06-24 15:01:32 +01:00 |
|
|
|
1b5f892d73
|
Add a currently wrong test for narrow.
|
2023-06-24 08:50:37 +01:00 |
|
|
|
b4653e41be
|
Helper function to build 3d arrays.
|
2023-06-24 06:29:06 +01:00 |
|
|
|
ae5dc5fbc6
|
Softmax tests + fix.
|
2023-06-23 22:46:36 +01:00 |
|
|
|
132859df75
|
Add some transpose tests.
|
2023-06-23 17:49:53 +01:00 |
|
|
|
fcb4e6b84f
|
Use a reference for the device.
|
2023-06-21 19:55:57 +01:00 |
|
|
|
7adffafeda
|
Abstract the gradient storage.
|
2023-06-21 14:29:48 +01:00 |
|
|
|
0839954770
|
Add some binary ops.
|
2023-06-21 08:32:35 +01:00 |
|
|
|
98b423145a
|
Bugfix for the contiguous strides.
|
2023-06-20 13:35:07 +01:00 |
|
|
|
f5b0aa815a
|
Get the addition/multiplication to work.
|
2023-06-20 11:07:59 +01:00 |
|
|
|
6c5fc767a8
|
Add the slice indexing.
|
2023-06-20 10:50:58 +01:00 |
|
|
|
7a31ba93e4
|
Start adding some ops.
|
2023-06-20 08:41:19 +01:00 |
|
|
|
26d6288eb6
|
Add an easy way to create tensor objects.
|
2023-06-19 20:59:26 +01:00 |
|
|
|
01eeb0e72f
|
Shuffle the shape bits around.
|
2023-06-19 20:22:12 +01:00 |
|
|
|
634e0c88ae
|
Add some basic test.
|
2023-06-19 19:50:17 +01:00 |
|
