Crate caffe2op_args

Structs

ArgMaxReducer
| Retrieve the argmax of an axis dimension | specified by the axis argument. | | Given an input tensor and two arguments | (axis and keepdims), returns a | tensor containing the indices of the | largest element along the given axis. | | If the keepdims arg is True (default), | the shape of the output tensor matches | the input tensor except the axis dimension | equals 1. | | Else, the axis dimension of the output | tensor is removed. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/arg_ops.cc |
ArgMinReducer
| Retrieve the argmin of an axis dimension | specified by the axis argument. | | Given an input tensor and two arguments | (axis and keepdims), returns a | tensor containing the indices of the | smallest element along the given axis. | | If the keepdims arg is True (default), | the shape of the output tensor matches | the input tensor except the axis dimension | equals 1. | | Else, the axis dimension of the output | tensor is removed. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/arg_ops.cc |
ArgOp