frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug correction: consider accreal in torch pkg

Browse Source
This commit is contained in:
Ronan Collobert
2012-03-07 17:18:11 +01:00
parent 007afa9988
commit 46e016be71
1 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
TensorMath.lua
View File

@ -249,11 +249,20 @@ local reals = {ByteTensor='unsigned char',
FloatTensor='float',
DoubleTensor='double'}
local accreals = {ByteTensor='long',
CharTensor='long',
ShortTensor='long',
IntTensor='long',
LongTensor='long',
FloatTensor='double',
DoubleTensor='double'}
for _,Tensor in ipairs({"ByteTensor", "CharTensor",
"ShortTensor", "IntTensor", "LongTensor",
"FloatTensor", "DoubleTensor"}) do
local real = reals[Tensor]
local accreal = accreals[Tensor]
function interface.luaname2wrapname(self, name)
return string.format('torch_%s_%s', Tensor, name)
@ -302,7 +311,7 @@ for _,Tensor in ipairs({"ByteTensor", "CharTensor",
cname("dot"),
{{name=Tensor},
{name=Tensor},
{name=real, creturned=true}})
{name=accreal, creturned=true}})
wrap("add",
cname("add"),
@ -480,7 +489,7 @@ for _,Tensor in ipairs({"ByteTensor", "CharTensor",
wrap("sum",
cname("sumall"),
{{name=Tensor},
{name=real, creturned=true}},
{name=accreal, creturned=true}},
cname("sum"),
{{name=Tensor, default=true, returned=true},
{name=Tensor},
@ -501,7 +510,7 @@ for _,Tensor in ipairs({"ByteTensor", "CharTensor",
wrap("trace",
cname("trace"),
{{name=Tensor},
{name=real, creturned=true}})
{name=accreal, creturned=true}})
wrap("cross",
cname("cross"),
@ -815,7 +824,7 @@ static void THTensor_random1__(THTensor *self, long b)
wrap("mean",
cname("meanall"),
{{name=Tensor},
{name=real, creturned=true}},
{name=accreal, creturned=true}},
cname("mean"),
{{name=Tensor, default=true, returned=true},
{name=Tensor},
@ -825,7 +834,7 @@ static void THTensor_random1__(THTensor *self, long b)
wrap(name,
cname(name .. "all"),
{{name=Tensor},
{name=real, creturned=true}},
{name=accreal, creturned=true}},
cname(name),
{{name=Tensor, default=true, returned=true},
{name=Tensor},
@ -844,7 +853,7 @@ static void THTensor_random1__(THTensor *self, long b)
cname("normall"),
{{name=Tensor},
{name=real, default=2},
{name=real, creturned=true}},
{name=accreal, creturned=true}},
cname("norm"),
{{name=Tensor, default=true, returned=true},
{name=Tensor},
@ -856,7 +865,7 @@ static void THTensor_random1__(THTensor *self, long b)
{{name=Tensor},
{name=Tensor},
{name=real, default=2},
{name=real, creturned=true}})
{name=accreal, creturned=true}})
wrap("linspace",
cname("linspace"),