提交 5b883789 编写于 作者: L Liufang Sang 提交者: Bai Yifan

add shape description in accuracy api doc (#22011)

* add shape description in accuracy api doc test=develop

* fix details test=develop

* fix details test=develop
上级 d4832077
......@@ -40,7 +40,8 @@ def accuracy(input, label, k=1, correct=None, total=None):
Args:
input(Variable): The input of accuracy layer, which is the predictions of network. A LoDTensor or Tensor with type float32,float64.
label(Variable): The label of dataset. LoDTensor or Tensor with type int32,int64.
The shape is ``[sample_number, class_dim]`` .
label(Variable): The label of dataset. LoDTensor or Tensor with type int32,int64. The shape is ``[sample_number, 1]`` .
k(int): The top k predictions for each class will be checked. Data type is int64 or int32.
correct(Variable): The correct predictions count. A Tensor with type int64 or int32.
total(Variable): The total entries count. A tensor with type int64 or int32.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册