提交 5fde8166 authored 作者: Frederic Bastien's avatar Frederic Bastien

added fct GpuSoftmax.infer_Shape

上级 979133cf
...@@ -299,6 +299,8 @@ class GpuSoftmax (Op): ...@@ -299,6 +299,8 @@ class GpuSoftmax (Op):
return self.__class__.__name__ return self.__class__.__name__
def make_node(self, x): def make_node(self, x):
return Apply(self, [x],[x.type()]) return Apply(self, [x],[x.type()])
def infer_shape(self, node, shape):
return shape
def c_code_cache_version(self): def c_code_cache_version(self):
#return () #return ()
return (1,) + inline_softmax.code_version return (1,) + inline_softmax.code_version
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论