mindspore 1.10.0__cp37-none-any.whl → 2.0.0rc1__cp37-none-any.whl

mindspore/ops/function/__init__.py

@@ -25,6 +25,7 @@ from . import (
     math_func,
     nn_func,
     linalg_func,
+    clip_func,
 )
 from .array_func import (
     unique,
@@ -34,29 +35,41 @@ 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,
     unstack,
     tensor_slice,
+    strided_slice,
     slice,
     scalar_to_array,
     scalar_to_tensor,
     tuple_to_array,
     expand_dims,
     squeeze,
+    unsqueeze,
     transpose,
     scatter_nd,
     scatter_nd_add,
@@ -65,12 +78,16 @@ 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,
     tensor_scatter_add,
     tensor_scatter_sub,
     tensor_scatter_mul,
@@ -79,6 +96,7 @@ from .array_func import (
     tensor_scatter_max,
     tensor_scatter_min,
     tensor_scatter_elements,
+    scatter,
     scatter_add,
     scatter_mul,
     scatter_max,
@@ -92,27 +110,55 @@ from .array_func import (
     space_to_batch_nd,
     batch_to_space_nd,
     range,
+    arange,
     select,
     one_hot,
     matrix_diag,
     matrix_diag_part,
     matrix_set_diag,
     diag,
+    diagflat,
     masked_select,
+    where,
     meshgrid,
     affine_grid,
     fills,
     broadcast_to,
     unsorted_segment_sum,
-    adaptive_max_pool2d,
     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,
@@ -126,14 +172,20 @@ from .math_func import (
     absolute,
     abs,
     argmin,
+    angle,
+    bincount,
     tensor_add,
+    cosine_similarity,
+    cov,
     add,
     addcdiv,
     addcmul,
     neg_tensor,
     neg,
+    negative,
     tensor_lt,
     less,
+    lt,
     tensor_le,
     le,
     lerp,
@@ -144,16 +196,27 @@ from .math_func import (
     tensor_ge,
     ge,
     tensor_sub,
+    rsqrt,
+    reciprocal,
+    real,
     sub,
+    subtract,
     sqrt,
     square,
     tensor_mul,
     mul,
+    multiply,
+    digamma,
+    lgamma,
     tensor_div,
     div,
+    divide,
+    true_divide,
     tensor_floordiv,
     floor_div,
     floordiv,
+    float_power,
+    fmod,
     xdivy,
     tensor_pow,
     pow,
@@ -163,38 +226,53 @@ 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,
     hypot,
     heaviside,
-    same_type_shape,
     gcd,
     log,
+    logdet,
     log_matrix_determinant,
+    slogdet,
     matrix_determinant,
+    det,
     linspace,
     matrix_solve,
     maximum,
     median,
+    nan_to_num,
+    nansum,
     logaddexp,
     logaddexp2,
     logit,
     std,
     ldexp,
     mv,
+    addbmm,
     addmv,
+    addmm,
     addr,
+    adjoint,
     inplace_add,
     inplace_sub,
     inplace_update,
     inv,
+    inverse,
     invert,
     minimum,
     renorm,
@@ -204,26 +282,45 @@ from .math_func import (
     logical_and,
     logsumexp,
     outer,
+    sign,
+    signbit,
+    sgn,
     sin,
+    sinc,
     cos,
+    t,
     tan,
     asin,
     acos,
+    arccos,
     atan,
+    arcsin,
+    arccosh,
+    arctan,
+    arctan2,
     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,
     ceil,
+    positive,
+    numel,
+    permute,
+    i0,
     bernoulli,
     bessel_i0,
     bessel_i0e,
@@ -241,15 +338,19 @@ from .math_func import (
     exp2,
     deg2rad,
     isreal,
+    is_complex,
     rad2deg,
     truncate_div,
     truncate_mod,
     trunc,
     gumbel_softmax,
+    kaiser_window,
     matmul,
+    inner,
     baddbmm,
     cummin,
     cummax,
+    cumsum,
     amin,
     amax,
     mean,
@@ -257,6 +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,
@@ -267,56 +380,141 @@ from .math_func import (
     rot90,
     remainder,
     iou,
-    bmm
+    bmm,
+    trapz,
+    cholesky,
+    cholesky_inverse,
+    conj,
+    cross,
+    erfinv,
+    less_equal,
+    cumprod,
+    greater,
+    greater_equal,
+    igamma,
+    igammac,
+    isinf,
+    logical_xor,
+    imag,
+    roll,
+    orgqr,
+    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,
     adaptive_avg_pool2d,
     adaptive_avg_pool3d,
     adaptive_max_pool1d,
+    adaptive_max_pool2d,
     adaptive_max_pool3d,
+    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,
     deformable_conv2d,
     fast_gelu,
+    flip,
+    fliplr,
+    flipud,
+    fractional_max_pool2d,
+    fractional_max_pool3d,
+    pixel_shuffle,
+    pixel_unshuffle,
     hardshrink,
     soft_shrink,
+    is_floating_point,
     intopk,
     interpolate,
+    upsample,
     kl_div,
     log_softmax,
     lrn,
     mish,
+    margin_ranking_loss,
+    max_unpool1d,
+    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,
+    multilabel_margin_loss,
+    multilabel_soft_margin_loss,
+    elu,
+    gelu,
+    hinge_embedding_loss,
+    gaussian_nll_loss,
+    lp_pool1d,
+    lp_pool2d,
+    mse_loss,
+    triplet_margin_loss,
+    msort,
+    channel_shuffle,
+    hardsigmoid,
 )
 from .linalg_func import (
+    eig,
+    geqrf,
     svd,
+    pinv,
+    qr
 )
 from .sparse_func import (
     coalesce,
@@ -328,6 +526,7 @@ from .sparse_func import (
     csr_gather,
     csr_mul,
     csr_mv,
+    csr_mm,
     csr_reduce_sum,
     csr_to_coo,
     csr2coo,
@@ -341,12 +540,14 @@ from .sparse_func import (
     make_coo_tensor,
     make_csr_tensor,
     make_row_tensor,
+    make_row_tensor_inner,
+    make_map_parameter,
     row_tensor_get_values,
     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,
@@ -359,21 +560,129 @@ from .random_func import (
     random_gamma,
     uniform_candidate_sampler,
     random_poisson,
-    random_shuffle,
+    log_uniform_candidate_sampler,
+    shuffle,
+    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,
     grad,
     value_and_grad,
+    jacfwd,
+    jacrev,
     jet,
     derivative,
     jvp,
     vjp,
-    linearize
+    linearize,
+    stop_gradient,
+    get_grad
 )
 from .debug_func import (
     print_,
 )
+from .image_func import (
+    bounding_box_decode,
+    bounding_box_encode,
+    check_valid,
+    crop_and_resize
+)
+from .spectral_func import (
+    blackman_window,
+    bartlett_window,
+)
+from .vmap_func import (
+    vmap,
+    _VmapGeneralPreprocess,
+    _VmapGeneralRule,
+)
+from .sparse_unary_func import (
+    csr_cos,
+    csr_tan,
+    csr_inv,
+    csr_exp,
+    csr_relu,
+    csr_expm1,
+    csr_isfinite,
+    csr_asin,
+    csr_sqrt,
+    csr_log,
+    csr_isnan,
+    csr_acos,
+    csr_floor,
+    csr_atan,
+    csr_square,
+    csr_relu6,
+    csr_sinh,
+    csr_ceil,
+    csr_cosh,
+    csr_softsign,
+    csr_log1p,
+    csr_round,
+    csr_tanh,
+    csr_neg,
+    csr_asinh,
+    csr_acosh,
+    csr_abs,
+    csr_isinf,
+    csr_atanh,
+    csr_sigmoid,
+    csr_sin,
+    coo_cos,
+    coo_tan,
+    coo_inv,
+    coo_exp,
+    coo_relu,
+    coo_expm1,
+    coo_isfinite,
+    coo_asin,
+    coo_sqrt,
+    coo_log,
+    coo_isnan,
+    coo_acos,
+    coo_floor,
+    coo_atan,
+    coo_square,
+    coo_relu6,
+    coo_sinh,
+    coo_ceil,
+    coo_cosh,
+    coo_softsign,
+    coo_log1p,
+    coo_round,
+    coo_tanh,
+    coo_neg,
+    coo_asinh,
+    coo_acosh,
+    coo_abs,
+    coo_isinf,
+    coo_atanh,
+    coo_sigmoid,
+    coo_sin
+)
+from .clip_func import (
+    clip_by_value,
+    clamp,
+    clip,
+    clip_by_global_norm,
+)
+from .other_func import (
+    depend,
+    partial,
+)
 
 __all__ = []
 __all__.extend(array_func.__all__)
@@ -385,4 +694,10 @@ __all__.extend(sparse_func.__all__)
 __all__.extend(random_func.__all__)
 __all__.extend(grad_func.__all__)
 __all__.extend(debug_func.__all__)
+__all__.extend(image_func.__all__)
+__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()