Something went wrong. Try again.
lol
Something went wrong. Try again.
C++
at master
123456789101112131415161718192021#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));}