646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
21 lines
301 B
C++
21 lines
301 B
C++
#undef NDEBUG
|
|
#include <cassert>
|
|
|
|
#include <iostream>
|
|
|
|
#include <torch/torch.h>
|
|
|
|
int main() {
|
|
torch::Tensor tensor = torch::eye(3);
|
|
|
|
float checkData[] = {
|
|
1, 0, 0,
|
|
0, 1, 0,
|
|
0, 0, 1
|
|
};
|
|
|
|
torch::Tensor check = torch::from_blob(checkData, {3, 3});
|
|
|
|
assert(tensor.allclose(check));
|
|
}
|