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