mindspore 2.0.0a0__cp38-cp38-win_amd64.whl → 2.0.0rc1__cp38-cp38-win_amd64.whl

mindspore/ops/function/__init__.py

@@ -35,20 +35,28 @@ from .array_func import (
     padding,
     fill,
     fill_,
+    full,
+    full_like,
+    chunk,
     tile,
     size,
     ones,
     ones_like,
+    zeros,
+    zeros_like,
     shape,
     shape_,
     ger,
     dyn_shape,
     rank,
+    hamming_window,
+    ravel,
     reshape,
     reshape_,
     reverse,
     reverse_sequence,
     flatten,
+    cat,
     concat,
     stack,
     unbind,
@@ -56,6 +64,7 @@ from .array_func import (
     tensor_slice,
     strided_slice,
     slice,
+    scalar_to_array,
     scalar_to_tensor,
     tuple_to_array,
     expand_dims,
@@ -69,10 +78,13 @@ from .array_func import (
     scatter_nd_div,
     scatter_nd_max,
     scatter_nd_min,
+    tril,
+    triu,
     gather,
     gather_d,
     gather_elements,
     gather_nd,
+    is_tensor,
     scalar_cast,
     masked_fill,
     narrow,
@@ -84,6 +96,7 @@ from .array_func import (
     tensor_scatter_max,
     tensor_scatter_min,
     tensor_scatter_elements,
+    scatter,
     scatter_add,
     scatter_mul,
     scatter_max,
@@ -104,7 +117,9 @@ from .array_func import (
     matrix_diag_part,
     matrix_set_diag,
     diag,
+    diagflat,
     masked_select,
+    where,
     meshgrid,
     affine_grid,
     fills,
@@ -112,18 +127,38 @@ from .array_func import (
     unsorted_segment_sum,
     col2im,
     split,
+    tensor_split,
+    vsplit,
+    hsplit,
+    dsplit,
     index_fill,
+    index_select,
     max,
     argmax,
     min,
     population_count,
-    top_k,
+    topk,
     expand,
     fold,
     unfold,
     diagonal,
     lstsq,
     mvlgamma,
+    argsort,
+    swapaxes,
+    swapdims,
+    sequence_mask,
+    repeat_elements,
+    repeat_interleave,
+    argwhere,
+    column_stack,
+    hstack,
+    movedim,
+    moveaxis,
+    searchsorted,
+    aminmax,
+    sort,
+    top_k
 )
 from .parameter_func import (
     assign,
@@ -138,7 +173,10 @@ from .math_func import (
     abs,
     argmin,
     angle,
+    bincount,
     tensor_add,
+    cosine_similarity,
+    cov,
     add,
     addcdiv,
     addcmul,
@@ -147,6 +185,7 @@ from .math_func import (
     negative,
     tensor_lt,
     less,
+    lt,
     tensor_le,
     le,
     lerp,
@@ -157,6 +196,9 @@ from .math_func import (
     tensor_ge,
     ge,
     tensor_sub,
+    rsqrt,
+    reciprocal,
+    real,
     sub,
     subtract,
     sqrt,
@@ -164,6 +206,8 @@ from .math_func import (
     tensor_mul,
     mul,
     multiply,
+    digamma,
+    lgamma,
     tensor_div,
     div,
     divide,
@@ -171,6 +215,8 @@ from .math_func import (
     tensor_floordiv,
     floor_div,
     floordiv,
+    float_power,
+    fmod,
     xdivy,
     tensor_pow,
     pow,
@@ -180,12 +226,19 @@ from .math_func import (
     floormod,
     lcm,
     tensor_exp,
+    einsum,
+    view_as_real,
+    var,
+    var_mean,
+    std_mean,
     exp,
     tensor_expm1,
     expm1,
     equal,
     not_equal,
     ne,
+    isneginf,
+    isposinf,
     isfinite,
     isnan,
     isclose,
@@ -193,12 +246,17 @@ from .math_func import (
     heaviside,
     gcd,
     log,
+    logdet,
     log_matrix_determinant,
+    slogdet,
     matrix_determinant,
+    det,
     linspace,
     matrix_solve,
     maximum,
     median,
+    nan_to_num,
+    nansum,
     logaddexp,
     logaddexp2,
     logit,
@@ -214,6 +272,7 @@ from .math_func import (
     inplace_sub,
     inplace_update,
     inv,
+    inverse,
     invert,
     minimum,
     renorm,
@@ -223,8 +282,13 @@ from .math_func import (
     logical_and,
     logsumexp,
     outer,
+    sign,
+    signbit,
+    sgn,
     sin,
+    sinc,
     cos,
+    t,
     tan,
     asin,
     acos,
@@ -237,13 +301,18 @@ from .math_func import (
     sinh,
     cosh,
     tanh,
+    tanhshrink,
     asinh,
+    arcsinh,
     acosh,
     atanh,
+    arctanh,
     atan2,
     bitwise_and,
     bitwise_or,
     bitwise_xor,
+    bitwise_left_shift,
+    bitwise_right_shift,
     erf,
     erfc,
     cdist,
@@ -269,12 +338,15 @@ from .math_func import (
     exp2,
     deg2rad,
     isreal,
+    is_complex,
     rad2deg,
     truncate_div,
     truncate_mod,
     trunc,
     gumbel_softmax,
+    kaiser_window,
     matmul,
+    inner,
     baddbmm,
     cummin,
     cummax,
@@ -286,9 +358,18 @@ from .math_func import (
     all,
     any,
     sparse_segment_mean,
+    block_diag,
+    atleast_1d,
+    dstack,
+    diff,
     atleast_2d,
+    cartesian_prod,
+    atleast_3d,
     vstack,
+    combinations,
+    dist,
     copysign,
+    hann_window,
     log2,
     xlogy,
     log10,
@@ -317,7 +398,24 @@ from .math_func import (
     imag,
     roll,
     orgqr,
-    matrix_exp
+    sum,
+    matrix_power,
+    matrix_exp,
+    logspace,
+    diag_embed,
+    fmax,
+    fmin,
+    inplace_index_add,
+    lu_unpack,
+    nanquantile,
+    polar,
+    polygamma,
+    quantile,
+    tril_indices,
+    triu_indices,
+    histc,
+    nextafter,
+    trace
 )
 from .nn_func import (
     adaptive_avg_pool1d,
@@ -329,12 +427,14 @@ from .nn_func import (
     avg_pool1d,
     avg_pool2d,
     avg_pool3d,
+    max_pool2d,
     max_pool3d,
     batch_norm,
     celu,
     bias_add,
     binary_cross_entropy,
     binary_cross_entropy_with_logits,
+    cosine_embedding_loss,
     dropout1d,
     dropout2d,
     dropout3d,
@@ -352,6 +452,7 @@ from .nn_func import (
     is_floating_point,
     intopk,
     interpolate,
+    upsample,
     kl_div,
     log_softmax,
     lrn,
@@ -361,29 +462,41 @@ from .nn_func import (
     max_unpool2d,
     max_unpool3d,
     hardswish,
+    hardtanh,
+    huber_loss,
     softsign,
+    silu,
     selu,
     softmax,
+    softmin,
+    softshrink,
     pdist,
     pad,
     prelu,
     mirror_pad,
     nll_loss,
     smooth_l1_loss,
+    l1_loss,
+    threshold,
+    leaky_relu,
     cross_entropy,
     grid_sample,
     ctc_greedy_decoder,
     ctc_loss,
     dropout,
     conv3d_transpose,
+    conv1d,
     conv2d,
     sigmoid,
+    logsigmoid,
     relu,
     relu6,
+    rrelu,
     conv3d,
     glu,
     multi_margin_loss,
-    multi_label_margin_loss,
+    multilabel_margin_loss,
+    multilabel_soft_margin_loss,
     elu,
     gelu,
     hinge_embedding_loss,
@@ -391,10 +504,17 @@ from .nn_func import (
     lp_pool1d,
     lp_pool2d,
     mse_loss,
+    triplet_margin_loss,
+    msort,
+    channel_shuffle,
+    hardsigmoid,
 )
 from .linalg_func import (
+    eig,
+    geqrf,
     svd,
-    pinv
+    pinv,
+    qr
 )
 from .sparse_func import (
     coalesce,
@@ -426,8 +546,8 @@ from .sparse_func import (
     row_tensor_get_indices,
     row_tensor_get_dense_shape,
     row_tensor_add,
-    sparse_add,
-    sparse_concat,
+    coo_add,
+    coo_concat,
     csr_add,
     csr_softmax,
     csr_to_dense,
@@ -442,7 +562,20 @@ from .random_func import (
     random_poisson,
     log_uniform_candidate_sampler,
     shuffle,
-    choice_with_mask
+    choice_with_mask,
+    normal,
+    laplace,
+    gamma,
+    poisson,
+    multinomial,
+    rand,
+    rand_like,
+    randn,
+    randn_like,
+    randint,
+    randint_like,
+    multinomial_with_replacement,
+    randperm,
 )
 from .grad import (
     grad_func,
@@ -455,7 +588,8 @@ from .grad import (
     jvp,
     vjp,
     linearize,
-    stop_gradient
+    stop_gradient,
+    get_grad
 )
 from .debug_func import (
     print_,
@@ -472,6 +606,8 @@ from .spectral_func import (
 )
 from .vmap_func import (
     vmap,
+    _VmapGeneralPreprocess,
+    _VmapGeneralRule,
 )
 from .sparse_unary_func import (
     csr_cos,
@@ -539,6 +675,13 @@ from .sparse_unary_func import (
 )
 from .clip_func import (
     clip_by_value,
+    clamp,
+    clip,
+    clip_by_global_norm,
+)
+from .other_func import (
+    depend,
+    partial,
 )
 
 __all__ = []
@@ -556,4 +699,5 @@ __all__.extend(spectral_func.__all__)
 __all__.extend(vmap_func.__all__)
 __all__.extend(sparse_unary_func.__all__)
 __all__.extend(clip_func.__all__)
+__all__.extend(other_func.__all__)
 __all__.sort()