Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-spline-conv
Commits
083e3611
Commit
083e3611
authored
Apr 10, 2018
by
rusty1s
Browse files
added backwar basis try
parent
03f4110b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
32 additions
and
0 deletions
+32
-0
aten/TH/generic/THBasis.c
aten/TH/generic/THBasis.c
+32
-0
No files found.
aten/TH/generic/THBasis.c
View file @
083e3611
...
...
@@ -42,6 +42,38 @@ void THTensor_(cubicBasisForward)(THTensor *basis, THLongTensor *weightIndex, TH
void
THTensor_
(
linearBasisBackward
)(
THTensor
*
self
,
THTensor
*
gradBasis
,
THTensor
*
pseudo
,
THLongTensor
*
kernelSize
,
THByteTensor
*
isOpenSpline
)
{
THTensor_
(
fill
)(
self
,
0
);
real
*
selfData
=
THTensor_
(
data
)(
self
);
real
*
gradBasisData
=
THTensor_
(
data
)(
gradBasis
);
real
*
pseudoData
=
THTensor_
(
data
)(
pseudo
);
int64_t
*
kernelSizeData
=
THLongTensor_data
(
kernelSize
);
uint8_t
*
isOpenSplineData
=
THByteTensor_data
(
isOpenSpline
);
ptrdiff_t
e
,
d
,
s
;
int64_t
k
,
kMod
,
wi
,
wiOffset
;
real
b
,
v
;
real
g
;
for
(
e
=
0
;
e
<
THTensor_
(
size
)(
pseudo
,
0
);
e
++
)
{
for
(
d
=
0
;
d
<
THTensor_
(
size
)(
pseudo
,
1
);
d
++
)
{
int64_t
quotient
=
pow
(
2
,
d
);
/* printf("e = %i, d = %i, stride0 = %i, stride1 = %i \n", e, d, pseudo->stride[0], pseudo->stride[1]); */
for
(
s
=
0
;
s
<
THTensor_
(
size
)(
gradBasis
,
1
);
s
++
)
{
kMod
=
(
s
/
quotient
)
%
2
;
real
v
=
pseudoData
[
e
*
pseudo
->
stride
[
0
]
+
d
*
pseudo
->
stride
[
1
]];
v
*=
kernelSizeData
[
d
]
-
2
*
isOpenSplineData
[
d
];
v
-=
floor
(
v
);
v
=
-
1
+
kMod
+
kMod
;
// grad code
g
=
v
;
ptrdiff_t
d_it
;
for
(
d_it
=
0
;
d_it
<
d
;
d_it
++
)
{
}
}
}
/* selfData[e * self->stride[0] + d * self->stride[1]] = 1; */
}
}
void
THTensor_
(
quadraticBasisBackward
)(
THTensor
*
self
,
THTensor
*
gradBasis
,
THTensor
*
pseudo
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment