#include <cstdio>
#include <cstdlib>
int main(int argc, char* argv[])
{
try {
int device = argc > 1 ? atoi(argv[1]) : 0;
float h_buffer[] = {1, 4, 2, 5, 3, 6 };
array in(2, 3, h_buffer);
svd(u, s_vec, vt, in);
array s_mat = diag(s_vec, 0,
false);
array in_recon = matmul(u, s_mat, vt(
seq(2), span));
fprintf(stderr,
"%s\n", e.
what());
throw;
}
return 0;
}
A multi dimensional data container.
Definition array.h:27
Definition exception.h:20
virtual const char * what() const
Definition exception.h:34
seq is used to create seq for indexing af::array
Definition seq.h:46
AFAPI void setDevice(const int device)
Sets the current device.
Definition algorithm.h:15
#define af_print(...)
Definition util.h:141