reset

rusty1s · rusty1s · commit de9300905a12 · 2018-08-06T17:22:05.000+02:00
diff --git a/cpu/dim_apply.h b/cpu/dim_apply.h
@@ -17,7 +17,7 @@
     auto TENSOR3##_stride = TENSOR3.stride(DIM);                               \
                                                                                \
     auto dims = TENSOR1.dim();                                                 \
-    auto zeros = at::zeros(dims, torch::CPU(at::kLong));                       \
+    auto zeros = at::zeros(torch::CPU(at::kLong), {dims});                     \
     auto counter = zeros.data<int64_t>();                                      \
     bool has_finished = false;                                                 \
                                                                                \
@@ -76,7 +76,7 @@
     auto TENSOR4##_stride = TENSOR4.stride(DIM);                               \
                                                                                \
     auto dims = TENSOR1.dim();                                                 \
-    auto zeros = at::zeros(dims, torch::CPU(at::kLong));                       \
+    auto zeros = at::zeros(torch::CPU(at::kLong), {dims});                     \
     auto counter = zeros.data<int64_t>();                                      \
     bool has_finished = false;                                                 \
                                                                                \

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@`
`17`	`17`	`auto TENSOR3##_stride = TENSOR3.stride(DIM); \`
`18`	`18`	`\`
`19`	`19`	`auto dims = TENSOR1.dim(); \`
`20`		`- auto zeros = at::zeros(dims, torch::CPU(at::kLong)); \`
	`20`	`+ auto zeros = at::zeros(torch::CPU(at::kLong), {dims}); \`
`21`	`21`	`auto counter = zeros.data<int64_t>(); \`
`22`	`22`	`bool has_finished = false; \`
`23`	`23`	`\`
`@@ -76,7 +76,7 @@`
`76`	`76`	`auto TENSOR4##_stride = TENSOR4.stride(DIM); \`
`77`	`77`	`\`
`78`	`78`	`auto dims = TENSOR1.dim(); \`
`79`		`- auto zeros = at::zeros(dims, torch::CPU(at::kLong)); \`
	`79`	`+ auto zeros = at::zeros(torch::CPU(at::kLong), {dims}); \`
`80`	`80`	`auto counter = zeros.data<int64_t>(); \`
`81`	`81`	`bool has_finished = false; \`
`82`	`82`	`\`