Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

tensorplay.functional.abs tensorplay.functional.abs_tensorplay.functional.absolute tensorplay.functional.acos tensorplay.functional.acos_tensorplay.functional.acosh tensorplay.functional.acosh_tensorplay.functional.adaptive_avg_pool1d tensorplay.functional.adaptive_max_pool1d tensorplay.functional.add tensorplay.functional.addbmm tensorplay.functional.addcdiv tensorplay.functional.addcmul tensorplay.functional.addmm tensorplay.functional.addmv tensorplay.functional.addmv_tensorplay.functional.addr tensorplay.functional.adjoint tensorplay.functional.affine_grid_generator tensorplay.functional.airy_ai tensorplay.functional.alias_copy tensorplay.functional.all tensorplay.functional.allclose tensorplay.functional.alpha_dropout tensorplay.functional.alpha_dropout_tensorplay.functional.amax tensorplay.functional.amin tensorplay.functional.aminmax tensorplay.functional.angle tensorplay.functional.any tensorplay.functional.arange tensorplay.functional.arccos tensorplay.functional.arccos_tensorplay.functional.arccosh tensorplay.functional.arccosh_tensorplay.functional.arcsin tensorplay.functional.arcsin_tensorplay.functional.arcsinh tensorplay.functional.arcsinh_tensorplay.functional.arctan tensorplay.functional.arctan_tensorplay.functional.arctan2 tensorplay.functional.arctanh tensorplay.functional.arctanh_tensorplay.functional.argmax tensorplay.functional.argmin tensorplay.functional.argsort tensorplay.functional.argwhere tensorplay.functional.as_strided tensorplay.functional.as_strided_tensorplay.functional.as_strided_copy tensorplay.functional.as_strided_scatter tensorplay.functional.asin tensorplay.functional.asin_tensorplay.functional.asinh tensorplay.functional.asinh_tensorplay.functional.atan tensorplay.functional.atan_tensorplay.functional.atan2 tensorplay.functional.atanh tensorplay.functional.atanh_tensorplay.functional.atleast_1d tensorplay.functional.atleast_2d tensorplay.functional.atleast_3d tensorplay.functional.avg_pool1d tensorplay.functional.baddbmm tensorplay.functional.bartlett_window tensorplay.functional.batch_norm_backward_elemt tensorplay.functional.batch_norm_backward_reduce tensorplay.functional.batch_norm_elemt tensorplay.functional.batch_norm_gather_stats tensorplay.functional.batch_norm_gather_stats_with_counts tensorplay.functional.batch_norm_stats tensorplay.functional.batch_norm_update_stats tensorplay.functional.bernoulli tensorplay.functional.bessel_j0 tensorplay.functional.bessel_j1 tensorplay.functional.bessel_y0 tensorplay.functional.bessel_y1 tensorplay.functional.bilinear tensorplay.functional.bincount tensorplay.functional.binomial tensorplay.functional.bitwise_and tensorplay.functional.bitwise_left_shift tensorplay.functional.bitwise_not tensorplay.functional.bitwise_or tensorplay.functional.bitwise_right_shift tensorplay.functional.bitwise_xor tensorplay.functional.blackman_window tensorplay.functional.bmm tensorplay.functional.broadcast_to tensorplay.functional.bucketize tensorplay.functional.can_cast tensorplay.functional.cartesian_prod tensorplay.functional.cat tensorplay.functional.ccol_indices_copy tensorplay.functional.cdist tensorplay.functional.ceil tensorplay.functional.ceil_tensorplay.functional.celu_tensorplay.functional.chain_matmul tensorplay.functional.channel_shuffle tensorplay.functional.chebyshev_polynomial_t tensorplay.functional.chebyshev_polynomial_u tensorplay.functional.chebyshev_polynomial_v tensorplay.functional.chebyshev_polynomial_w tensorplay.functional.cholesky_inverse tensorplay.functional.cholesky_solve tensorplay.functional.choose_qparams_optimized tensorplay.functional.chunk tensorplay.functional.clamp tensorplay.functional.clamp_tensorplay.functional.clamp_max_tensorplay.functional.clamp_min_tensorplay.functional.clip tensorplay.functional.clip_tensorplay.functional.clone tensorplay.functional.col_indices_copy tensorplay.functional.column_stack tensorplay.functional.combinations tensorplay.functional.complex tensorplay.functional.concat tensorplay.functional.concatenate tensorplay.functional.conj tensorplay.functional.conj_physical tensorplay.functional.conj_physical_tensorplay.functional.conv_tbc tensorplay.functional.conv_transpose1d tensorplay.functional.conv_transpose2d tensorplay.functional.conv_transpose3d tensorplay.functional.conv1d tensorplay.functional.conv3d tensorplay.functional.convolution tensorplay.functional.copysign tensorplay.functional.corrcoef tensorplay.functional.cos tensorplay.functional.cos_tensorplay.functional.cosh tensorplay.functional.cosh_tensorplay.functional.cosine_embedding_loss tensorplay.functional.cosine_similarity tensorplay.functional.count_nonzero tensorplay.functional.cov tensorplay.functional.cross tensorplay.functional.crow_indices_copy tensorplay.functional.ctc_loss tensorplay.functional.cudnn_affine_grid_generator tensorplay.functional.cudnn_batch_norm tensorplay.functional.cudnn_convolution tensorplay.functional.cudnn_convolution_add_relu tensorplay.functional.cudnn_convolution_relu tensorplay.functional.cudnn_convolution_transpose tensorplay.functional.cudnn_grid_sampler tensorplay.functional.cudnn_is_acceptable tensorplay.functional.cummax tensorplay.functional.cummin tensorplay.functional.cumprod tensorplay.functional.cumsum tensorplay.functional.cumulative_trapezoid tensorplay.functional.deg2rad tensorplay.functional.deg2rad_tensorplay.functional.dequantize tensorplay.functional.det tensorplay.functional.detach tensorplay.functional.detach_tensorplay.functional.detach_copy tensorplay.functional.diag tensorplay.functional.diag_embed tensorplay.functional.diagflat tensorplay.functional.diagonal tensorplay.functional.diagonal_copy tensorplay.functional.diagonal_scatter tensorplay.functional.diff tensorplay.functional.digamma tensorplay.functional.dist tensorplay.functional.div tensorplay.functional.divide tensorplay.functional.dot tensorplay.functional.dropout_tensorplay.functional.dsplit tensorplay.functional.dstack tensorplay.functional.embedding tensorplay.functional.embedding_renorm_tensorplay.functional.empty_like tensorplay.functional.empty_permuted tensorplay.functional.empty_quantized tensorplay.functional.empty_strided tensorplay.functional.entr tensorplay.functional.eq tensorplay.functional.equal tensorplay.functional.erf tensorplay.functional.erf_tensorplay.functional.erfc tensorplay.functional.erfc_tensorplay.functional.erfcx tensorplay.functional.erfinv tensorplay.functional.exp tensorplay.functional.exp_tensorplay.functional.exp2 tensorplay.functional.exp2_tensorplay.functional.expand_copy tensorplay.functional.expm1 tensorplay.functional.expm1_tensorplay.functional.eye tensorplay.functional.fake_quantize_per_channel_affine tensorplay.functional.fake_quantize_per_tensor_affine tensorplay.functional.fbgemm_linear_fp16_weight tensorplay.functional.fbgemm_linear_fp16_weight_fp32_activation tensorplay.functional.fbgemm_linear_int8_weight tensorplay.functional.fbgemm_linear_int8_weight_fp32_activation tensorplay.functional.fbgemm_linear_quantize_weight tensorplay.functional.fbgemm_pack_gemm_matrix_fp16 tensorplay.functional.fbgemm_pack_quantized_matrix tensorplay.functional.feature_alpha_dropout tensorplay.functional.feature_alpha_dropout_tensorplay.functional.feature_dropout tensorplay.functional.feature_dropout_tensorplay.functional.fill tensorplay.functional.fill_tensorplay.functional.fix tensorplay.functional.fix_tensorplay.functional.flatten tensorplay.functional.flip tensorplay.functional.fliplr tensorplay.functional.flipud tensorplay.functional.float_power tensorplay.functional.floor tensorplay.functional.floor_tensorplay.functional.floor_divide tensorplay.functional.fmax tensorplay.functional.fmin tensorplay.functional.fmod tensorplay.functional.frac tensorplay.functional.frac_tensorplay.functional.frexp tensorplay.functional.from_file tensorplay.functional.full_like tensorplay.functional.fused_moving_avg_obs_fake_quant tensorplay.functional.gammainc tensorplay.functional.gammaincc tensorplay.functional.gather tensorplay.functional.gcd tensorplay.functional.gcd_tensorplay.functional.ge tensorplay.functional.geqrf tensorplay.functional.ger tensorplay.functional.get_device tensorplay.functional.greater tensorplay.functional.greater_equal tensorplay.functional.grid_sampler_2d tensorplay.functional.grid_sampler_3d tensorplay.functional.group_norm tensorplay.functional.gru tensorplay.functional.gru_cell tensorplay.functional.gt tensorplay.functional.hamming_window tensorplay.functional.hann_window tensorplay.functional.hardshrink tensorplay.functional.hash_tensor tensorplay.functional.heaviside tensorplay.functional.hermite_polynomial_h tensorplay.functional.hermite_polynomial_he tensorplay.functional.hinge_embedding_loss tensorplay.functional.histc tensorplay.functional.histogramdd tensorplay.functional.hsplit tensorplay.functional.hstack tensorplay.functional.hypot tensorplay.functional.i0 tensorplay.functional.i0_tensorplay.functional.i0e tensorplay.functional.i1 tensorplay.functional.i1e tensorplay.functional.igamma tensorplay.functional.igammac tensorplay.functional.imag tensorplay.functional.index_add tensorplay.functional.index_copy tensorplay.functional.index_put_tensorplay.functional.index_reduce tensorplay.functional.index_select tensorplay.functional.indices_copy tensorplay.functional.inner tensorplay.functional.instance_norm tensorplay.functional.int_repr tensorplay.functional.inverse tensorplay.functional.is_complex tensorplay.functional.is_conj tensorplay.functional.is_distributed tensorplay.functional.is_floating_point tensorplay.functional.is_inference tensorplay.functional.is_neg tensorplay.functional.is_nonzero tensorplay.functional.is_same_size tensorplay.functional.is_signed tensorplay.functional.is_vulkan_available tensorplay.functional.isclose tensorplay.functional.isfinite tensorplay.functional.isin tensorplay.functional.isinf tensorplay.functional.isnan tensorplay.functional.isneginf tensorplay.functional.isposinf tensorplay.functional.isreal tensorplay.functional.istft tensorplay.functional.kaiser_window tensorplay.functional.kl_div tensorplay.functional.kron tensorplay.functional.kthvalue tensorplay.functional.laguerre_polynomial_l tensorplay.functional.lcm tensorplay.functional.lcm_tensorplay.functional.ldexp tensorplay.functional.ldexp_tensorplay.functional.le tensorplay.functional.legendre_polynomial_p tensorplay.functional.lerp tensorplay.functional.less tensorplay.functional.less_equal tensorplay.functional.lgamma tensorplay.functional.linspace tensorplay.functional.log tensorplay.functional.log_tensorplay.functional.log_ndtr tensorplay.functional.log10 tensorplay.functional.log10_tensorplay.functional.log1p tensorplay.functional.log1p_tensorplay.functional.log2 tensorplay.functional.log2_tensorplay.functional.logaddexp tensorplay.functional.logaddexp2 tensorplay.functional.logcumsumexp tensorplay.functional.logdet tensorplay.functional.logical_and tensorplay.functional.logical_not tensorplay.functional.logical_or tensorplay.functional.logical_xor tensorplay.functional.logit tensorplay.functional.logit_tensorplay.functional.logspace tensorplay.functional.logsumexp tensorplay.functional.lstm tensorplay.functional.lstm_cell tensorplay.functional.lt tensorplay.functional.lu_solve tensorplay.functional.lu_unpack tensorplay.functional.margin_ranking_loss tensorplay.functional.masked_fill tensorplay.functional.masked_select tensorplay.functional.matmul tensorplay.functional.matrix_exp tensorplay.functional.matrix_power tensorplay.functional.max_pool1d tensorplay.functional.max_pool3d tensorplay.functional.maximum tensorplay.functional.mean tensorplay.functional.median tensorplay.functional.meshgrid tensorplay.functional.minimum tensorplay.functional.miopen_batch_norm tensorplay.functional.miopen_convolution tensorplay.functional.miopen_convolution_add_relu tensorplay.functional.miopen_convolution_relu tensorplay.functional.miopen_convolution_transpose tensorplay.functional.miopen_ctc_loss tensorplay.functional.miopen_depthwise_convolution tensorplay.functional.miopen_rnn tensorplay.functional.mkldnn_adaptive_avg_pool2d tensorplay.functional.mkldnn_convolution tensorplay.functional.mkldnn_linear_backward_weights tensorplay.functional.mkldnn_max_pool2d tensorplay.functional.mkldnn_max_pool3d tensorplay.functional.mkldnn_rnn_layer tensorplay.functional.mm tensorplay.functional.mode tensorplay.functional.modified_bessel_i0 tensorplay.functional.modified_bessel_i1 tensorplay.functional.modified_bessel_k0 tensorplay.functional.modified_bessel_k1 tensorplay.functional.moveaxis tensorplay.functional.movedim tensorplay.functional.msort tensorplay.functional.mul tensorplay.functional.multinomial tensorplay.functional.multiply tensorplay.functional.mv tensorplay.functional.mvlgamma tensorplay.functional.nan_to_num tensorplay.functional.nan_to_num_tensorplay.functional.nanmean tensorplay.functional.nanmedian tensorplay.functional.nansum tensorplay.functional.narrow tensorplay.functional.narrow_copy tensorplay.functional.native_batch_norm tensorplay.functional.native_channel_shuffle tensorplay.functional.native_group_norm tensorplay.functional.native_layer_norm tensorplay.functional.native_norm tensorplay.functional.ndtr tensorplay.functional.ndtri tensorplay.functional.ne tensorplay.functional.neg tensorplay.functional.neg_tensorplay.functional.negative tensorplay.functional.negative_tensorplay.functional.nextafter tensorplay.functional.nonzero tensorplay.functional.nonzero_static tensorplay.functional.norm tensorplay.functional.norm_except_dim tensorplay.functional.normal tensorplay.functional.not_equal tensorplay.functional.nuclear_norm tensorplay.functional.numel tensorplay.functional.ones_like tensorplay.functional.orgqr tensorplay.functional.ormqr tensorplay.functional.outer tensorplay.functional.pairwise_distance tensorplay.functional.pdist tensorplay.functional.permute tensorplay.functional.permute_copy tensorplay.functional.pinverse tensorplay.functional.pixel_unshuffle tensorplay.functional.poisson tensorplay.functional.poisson_nll_loss tensorplay.functional.polar tensorplay.functional.polygamma tensorplay.functional.positive tensorplay.functional.pow tensorplay.functional.prelu tensorplay.functional.prod tensorplay.functional.promote_types tensorplay.functional.put tensorplay.functional.q_per_channel_axis tensorplay.functional.q_per_channel_scales tensorplay.functional.q_per_channel_zero_points tensorplay.functional.q_scale tensorplay.functional.q_zero_point tensorplay.functional.quantize_per_channel tensorplay.functional.quantize_per_tensor tensorplay.functional.quantize_per_tensor_dynamic tensorplay.functional.quantized_batch_norm tensorplay.functional.quantized_gru_cell tensorplay.functional.quantized_lstm_cell tensorplay.functional.quantized_max_pool1d tensorplay.functional.quantized_max_pool2d tensorplay.functional.quantized_max_pool3d tensorplay.functional.quantized_rnn_relu_cell tensorplay.functional.quantized_rnn_tanh_cell tensorplay.functional.rad2deg tensorplay.functional.rad2deg_tensorplay.functional.rand_like tensorplay.functional.randint tensorplay.functional.randint_like tensorplay.functional.randn_like tensorplay.functional.randperm tensorplay.functional.range tensorplay.functional.ravel tensorplay.functional.real tensorplay.functional.reciprocal tensorplay.functional.reciprocal_tensorplay.functional.relu_tensorplay.functional.remainder tensorplay.functional.renorm tensorplay.functional.repeat_interleave tensorplay.functional.reshape tensorplay.functional.resize_as_tensorplay.functional.resize_as_sparse_tensorplay.functional.resolve_conj tensorplay.functional.resolve_neg tensorplay.functional.result_type tensorplay.functional.rms_norm tensorplay.functional.rnn_relu tensorplay.functional.rnn_relu_cell tensorplay.functional.rnn_tanh tensorplay.functional.rnn_tanh_cell tensorplay.functional.roll tensorplay.functional.rot90 tensorplay.functional.round tensorplay.functional.round_tensorplay.functional.row_indices_copy tensorplay.functional.row_stack tensorplay.functional.rrelu tensorplay.functional.rrelu_tensorplay.functional.rsqrt tensorplay.functional.rsqrt_tensorplay.functional.rsub tensorplay.functional.scalar_tensor tensorplay.functional.scaled_modified_bessel_k0 tensorplay.functional.scaled_modified_bessel_k1 tensorplay.functional.scatter tensorplay.functional.scatter_add tensorplay.functional.scatter_reduce tensorplay.functional.searchsorted tensorplay.functional.segment_reduce tensorplay.functional.select tensorplay.functional.select_copy tensorplay.functional.select_scatter tensorplay.functional.selu tensorplay.functional.selu_tensorplay.functional.sgn tensorplay.functional.shifted_chebyshev_polynomial_t tensorplay.functional.shifted_chebyshev_polynomial_u tensorplay.functional.shifted_chebyshev_polynomial_v tensorplay.functional.shifted_chebyshev_polynomial_w tensorplay.functional.sigmoid tensorplay.functional.sigmoid_tensorplay.functional.sign tensorplay.functional.signbit tensorplay.functional.sin tensorplay.functional.sin_tensorplay.functional.sinc tensorplay.functional.sinc_tensorplay.functional.sinh tensorplay.functional.sinh_tensorplay.functional.slice_copy tensorplay.functional.slice_inverse tensorplay.functional.slice_scatter tensorplay.functional.slogdet tensorplay.functional.softmax tensorplay.functional.sort tensorplay.functional.sparse_bsc_tensor tensorplay.functional.sparse_bsr_tensor tensorplay.functional.sparse_coo_tensor tensorplay.functional.sparse_csc_tensor tensorplay.functional.sparse_csr_tensor tensorplay.functional.spherical_bessel_j0 tensorplay.functional.split tensorplay.functional.split_copy tensorplay.functional.split_with_sizes_copy tensorplay.functional.sqrt tensorplay.functional.sqrt_tensorplay.functional.square tensorplay.functional.square_tensorplay.functional.squeeze tensorplay.functional.squeeze_copy tensorplay.functional.stack tensorplay.functional.std tensorplay.functional.std_mean tensorplay.functional.stft tensorplay.functional.sub tensorplay.functional.subtract tensorplay.functional.sum tensorplay.functional.svd tensorplay.functional.swapaxes tensorplay.functional.swapdims tensorplay.functional.sym_constrain_range tensorplay.functional.sym_constrain_range_for_size tensorplay.functional.t tensorplay.functional.t_copy tensorplay.functional.take tensorplay.functional.take_along_dim tensorplay.functional.tan tensorplay.functional.tan_tensorplay.functional.tanh tensorplay.functional.tanh_tensorplay.functional.tensor_split tensorplay.functional.threshold tensorplay.functional.threshold_tensorplay.functional.tile tensorplay.functional.topk tensorplay.functional.trace tensorplay.functional.transpose tensorplay.functional.transpose_copy tensorplay.functional.trapezoid tensorplay.functional.trapz tensorplay.functional.triangular_solve tensorplay.functional.tril tensorplay.functional.tril_indices tensorplay.functional.triplet_margin_loss tensorplay.functional.triu tensorplay.functional.triu_indices tensorplay.functional.true_divide tensorplay.functional.trunc tensorplay.functional.trunc_tensorplay.functional.unbind tensorplay.functional.unbind_copy tensorplay.functional.unflatten tensorplay.functional.unfold_copy tensorplay.functional.unique tensorplay.functional.unique_consecutive tensorplay.functional.unsqueeze tensorplay.functional.unsqueeze_copy tensorplay.functional.values_copy tensorplay.functional.vander tensorplay.functional.var tensorplay.functional.var_mean tensorplay.functional.vdot tensorplay.functional.view_as_complex tensorplay.functional.view_as_complex_copy tensorplay.functional.view_as_real tensorplay.functional.view_as_real_copy tensorplay.functional.view_copy tensorplay.functional.vsplit tensorplay.functional.vstack tensorplay.functional.where tensorplay.functional.xlog1py tensorplay.functional.xlogy tensorplay.functional.xlogy_tensorplay.functional.zero_tensorplay.functional.zeros_like tensorplay.functional.zeta

API symbolsfunctional

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

tensorplay.functional.diagflat

tensorplay.functional.diagflat(input, offset=0)[source]

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.functional.diag_embed

Previous Page

tensorplay.functional.diagonal

Next Page

On this page