提交 4f968fb0 authored 作者: Iban Harlouchet's avatar Iban Harlouchet

Added a docstring

上级 4765f30e
...@@ -1007,6 +1007,11 @@ def to_one_hot(y, nb_class, dtype=None): ...@@ -1007,6 +1007,11 @@ def to_one_hot(y, nb_class, dtype=None):
return ret return ret
class Unique(theano.Op): class Unique(theano.Op):
"""
Wraps numpy.unique.
This op is not implemented on the GPU.
"""
__props__ = ("return_index", "return_inverse", "return_counts") __props__ = ("return_index", "return_inverse", "return_counts")
def __init__(self, return_index=False, return_inverse=False, def __init__(self, return_index=False, return_inverse=False,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论