提交 9c404031 authored 作者: Frederic Bastien's avatar Frederic Bastien

added comment for how to get the numpy default version of MaxAndArgmax.

上级 3c089d9e
......@@ -1302,7 +1302,8 @@ class MaxAndArgmax(Op):
If axis is None it means to calculate the max over the last dimension which is
DIFFERENT FROM NUMPY!!
To have the behavior of numpy do a flatten of the input before passing the data to this op.
If the input to flatten is not ccontiguous, this will make a copy to a contiguous version.
"""
nin=2 # tensor, axis
nout=2 # max val, max idx
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论