forked from torch/cutorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTensor.lua
47 lines (43 loc) · 1.49 KB
/
Tensor.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
function torch.CudaTensor.apply(self, func)
local x = torch.FloatTensor(self:size()):copy(self)
x:apply(func)
self:copy(x)
end
local function Tensor__type(self,type)
local current = torch.typename(self)
if not type then return current end
if type ~= current then
local new = torch.getmetatable(type).new()
if self:nElement() > 0 then
new:resize(self:size()):copy(self)
end
return new
else
return self
end
end
local function Tensor__typeAs(self,tensor)
return self:type(tensor:type())
end
local function Tensor__cuda(self,type)
return self:type('torch.CudaTensor')
end
local function Tensor__double(self,type)
return self:type('torch.DoubleTensor')
end
local function Tensor__float(self,type)
return self:type('torch.FloatTensor')
end
rawset(torch.getmetatable('torch.DoubleTensor'), 'cuda', Tensor__cuda)
rawset(torch.getmetatable('torch.FloatTensor'), 'cuda', Tensor__cuda)
rawset(torch.getmetatable('torch.CudaTensor'), 'cuda', Tensor__cuda)
rawset(torch.getmetatable('torch.CudaTensor'), 'type', Tensor__type)
rawset(torch.getmetatable('torch.CudaTensor'), 'typeAs', Tensor__typeAs)
rawset(torch.getmetatable('torch.CudaTensor'), 'double', Tensor__double)
rawset(torch.getmetatable('torch.CudaTensor'), 'float', Tensor__float)
do
local metatable = torch.getmetatable('torch.CudaTensor')
for _,func in pairs{'expand', 'expandAs', 'view', 'viewAs', 'repeatTensor', 'permute'} do
rawset(metatable, func, torch[func])
end
end