oneflow.rst

oneflow
===================================
oneflow
----------------------------------
.. currentmodule:: oneflow
.. automodule:: oneflow
    :members: adaptive_avg_pool1d, 
            adaptive_avg_pool2d, 
            adaptive_avg_pool3d, 
            acos, 
            add, 
            acosh, 
            addmm, 
            arcsin, 
            arcsinh, 
            arccosh, 
            arctan, 
            arctanh, 
            argmax, 
            arange, 
            argsort, 
            argwhere,
            asin,  
            asinh, 
            atan, 
            atan2, 
            atanh, 
            bernoulli,
            cat, 
            cast, 
            ceil, 
            chunk, 
            clamp, 
            clip, 
            cos, 
            cosh, 
            diag, 
            div, 
            eq, 
            equal, 
            expand, 
            eye,
            exp, 
            expm1, 
            erf, 
            erfc, 
            flatten, 
            flip, 
            floor, 
            fmod, 
            gather, 
            gather_nd, 
            gelu, 
            gt, 
            in_top_k, 
            index_select,
            logical_and,
            logical_or,
            logical_xor,
            load, 
            log, 
            log1p, 
            lt, 
            masked_fill, 
            masked_select, 
            matmul, 
            narrow, 
            max, 
            mean,
            mish,  
            min, 
            meshgrid, 
            mul, 
            neg, 
            negative, 
            nonzero,
            empty,
            ones, 
            ones_like, 
            pow,
            prod,  
            rand,
            randn,
            repeat, 
            reshape, 
            randperm,
            reciprocal,
            round,  
            save, 
            scatter,
            scatter_add,
            scatter_nd, 
            sin, 
            sinh, 
            sign, 
            selu, 
            silu, 
            slice, 
            slice_update, 
            softsign, 
            sort, 
            softplus, 
            sigmoid, 
            softmax, 
            squeeze, 
            split, 
            stack, 
            std,
            sub, 
            sum, 
            sqrt, 
            square,  
            tan, 
            tanh, 
            tensor, 
            tile, 
            to, 
            transpose, 
            unsqueeze, 
            var, 
            view, 
            where, 
            zeros, 
            zeros_like,
            is_nonzero,
            no_grad,
            grad_enable,
            inference_mode,
            is_grad_enabled,
.. autofunction:: oneflow.data.load_mnist(train_batch_size=100, test_batch_size=100, data_format='NCHW')
.. autofunction:: oneflow.relu