tan docs bug

This commit is contained in:
ling 2022-07-15 14:05:13 +08:00
parent 3412fa4366
commit 3adf260a91
2 changed files with 3 additions and 2 deletions

View File

@ -1666,6 +1666,7 @@ mindspore.Tensor
.. py:method:: tan()
返回每个元素的正切值。
.. math::
out_i = tan(x_i)

View File

@ -106,8 +106,8 @@ TypePtr TensorScatterArithmeticInferType(const PrimitivePtr &primitive,
return CheckAndConvertUtils::CheckTensorTypeSame(type_dict, common_valid_types_with_bool, prim_name);
}
/* tensor scatter div grad need div
* div do not support int8 and uint8 currently */
/* tensor scatter div grad need Neg
* Neg do not support int8 and uint8 currently */
if (prim_name == prim::kPrimTensorScatterDiv->name()) {
const std::set<TypePtr> tensor_scatter_div_data_type = {kInt16, kInt32, kInt64, kUInt16, kUInt32,
kUInt64, kFloat16, kFloat32, kFloat64};