X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=svrt.c;h=307fcf646ef839f69bf4f3f8613a68e21682554d;hp=1a2449b04ee43638b9bc961478dbbe5db3adfabe;hb=034e1a7f3e7174a2eebfcff388fa0fefc8c331e7;hpb=24368498f3065e8a4be34c5e8e2b68f9d1220f7d

diff --git a/svrt.c b/svrt.c
index 1a2449b..307fcf6 100644
--- a/svrt.c
+++ b/svrt.c
@@ -35,6 +35,11 @@ THByteTensor *generate_vignettes(long n_problem, THLongTensor *labels) {
   long *m, *l;
   unsigned char *a, *b;
 
+  if(THLongTensor_nDimension(labels) != 1) {
+    printf("Label tensor has to be of dimension 1.\n");
+    exit(1);
+  }
+
   nb_vignettes = THLongTensor_size(labels, 0);
   m = THLongTensor_storage(labels)->data + THLongTensor_storageOffset(labels);
   st0 = THLongTensor_stride(labels, 0);