提交 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): ...@@ -1302,7 +1302,8 @@ class MaxAndArgmax(Op):
If axis is None it means to calculate the max over the last dimension which is If axis is None it means to calculate the max over the last dimension which is
DIFFERENT FROM NUMPY!! 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 nin=2 # tensor, axis
nout=2 # max val, max idx nout=2 # max val, max idx
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论