def forward_gpu(self, x):
y = cuda.elementwise(
'T x', 'T y',
'y = x >= 0 ? 1 : -1', 'bst_fwd')(
x[0])
return y,
微信
公众号
扫码关注公众号