Commit 5515f9e 1 parent 8121ec6 commit 5515f9e Copy full SHA for 5515f9e
File tree 2 files changed +2
-2
lines changed
2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -234,7 +234,7 @@ int torch_tensor_get_rank(const torch_tensor_t tensor)
234
234
return t->sizes ().size ();
235
235
}
236
236
237
- const long int * torch_tensor_get_sizes (const torch_tensor_t tensor)
237
+ const long long int * torch_tensor_get_sizes (const torch_tensor_t tensor)
238
238
{
239
239
auto t = reinterpret_cast <torch::Tensor*>(tensor);
240
240
return t->sizes ().data ();
Original file line number Diff line number Diff line change @@ -125,7 +125,7 @@ EXPORT_C int torch_tensor_get_rank(const torch_tensor_t tensor);
125
125
* @param Torch Tensor to determine the rank of
126
126
* @return pointer to the sizes array of the Torch Tensor
127
127
*/
128
- EXPORT_C const long int * torch_tensor_get_sizes (const torch_tensor_t tensor );
128
+ EXPORT_C const long long int * torch_tensor_get_sizes (const torch_tensor_t tensor );
129
129
130
130
/**
131
131
* Function to delete a Torch Tensor to clean up
You can’t perform that action at this time.
0 commit comments