TensorPlay

Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

Functions 605

#

arange

functionFull reference ↗
tensorplay.functional.arange(*args, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

argsort

functionFull reference ↗
tensorplay.functional.argsort(input, dim=-1, descending=False, *, stable=<object object>, out=None)[source]
#

as_strided_scatter

functionFull reference ↗
tensorplay.functional.as_strided_scatter(input, src, size, stride, storage_offset=None)[source]
#

avg_pool1d

functionFull reference ↗
tensorplay.functional.avg_pool1d(input, kernel_size, stride=[], padding=0, ceil_mode=False, count_include_pad=True)[source]
#

batch_norm_backward_elemt

functionFull reference ↗
tensorplay.functional.batch_norm_backward_elemt(grad_out, input, mean, invstd, weight, sum_dy, sum_dy_xmu, count)[source]
#

batch_norm_backward_reduce

functionFull reference ↗
tensorplay.functional.batch_norm_backward_reduce(grad_out, input, mean, invstd, weight, input_g, weight_g, bias_g)[source]
#

batch_norm_elemt

functionFull reference ↗
tensorplay.functional.batch_norm_elemt(input, weight, bias, mean, invstd, eps, *, out=None)[source]
#

batch_norm_gather_stats_with_counts

functionFull reference ↗
tensorplay.functional.batch_norm_gather_stats_with_counts(input, mean, invstd, running_mean, running_var, momentum, eps, counts)[source]
#

batch_norm_gather_stats

functionFull reference ↗
tensorplay.functional.batch_norm_gather_stats(input, mean, invstd, running_mean, running_var, momentum, eps, count)[source]
#

batch_norm_update_stats

functionFull reference ↗
tensorplay.functional.batch_norm_update_stats(input, running_mean, running_var, momentum)[source]
#

bucketize

functionFull reference ↗
tensorplay.functional.bucketize(input, boundaries, *, out_int32=False, right=False, out=None)[source]
#

choose_qparams_optimized

functionFull reference ↗
tensorplay.functional.choose_qparams_optimized(input, numel, n_bins, ratio, bit_width)[source]
#

conv_transpose1d

functionFull reference ↗
tensorplay.functional.conv_transpose1d(input, weight, bias=None, stride=[], padding=[], output_padding=[], groups=1, dilation=[])[source]
#

conv_transpose2d

functionFull reference ↗
tensorplay.functional.conv_transpose2d(input, weight, bias=None, stride=[1, 1], padding=[0, 0], output_padding=[0, 0], groups=1, dilation=[1, 1])[source]
#

conv_transpose3d

functionFull reference ↗
tensorplay.functional.conv_transpose3d(input, weight, bias=None, stride=[1, 1, 1], padding=[0, 0, 0], output_padding=[0, 0, 0], groups=1, dilation=[1, 1, 1])[source]
#

conv1d

functionFull reference ↗
tensorplay.functional.conv1d(input, weight, bias=None, stride=[1], padding=[0], dilation=[1], groups=1)[source]
#

conv3d

functionFull reference ↗
tensorplay.functional.conv3d(input, weight, bias=None, stride=[1, 1, 1], padding=[0, 0, 0], dilation=[1, 1, 1], groups=1)[source]
#

convolution

functionFull reference ↗
tensorplay.functional.convolution(input, weight, bias, stride, padding, dilation, transposed, output_padding, groups)[source]
#

ctc_loss

functionFull reference ↗
tensorplay.functional.ctc_loss(log_probs, targets, input_lengths, target_lengths, blank=0, reduction=1, zero_infinity=False)[source]
#

cudnn_batch_norm

functionFull reference ↗
tensorplay.functional.cudnn_batch_norm(input, weight, bias, running_mean, running_var, training, exponential_average_factor, epsilon, *, out=None)[source]
#

cudnn_convolution_add_relu

functionFull reference ↗
tensorplay.functional.cudnn_convolution_add_relu(input, weight, z, alpha, bias, stride, padding, dilation, groups)[source]
#

cudnn_convolution_relu

functionFull reference ↗
tensorplay.functional.cudnn_convolution_relu(input, weight, bias, stride, padding, dilation, groups)[source]
#

cudnn_convolution_transpose

functionFull reference ↗
tensorplay.functional.cudnn_convolution_transpose(input, weight, padding, output_padding, stride, dilation, groups, benchmark, deterministic, allow_tf32)[source]
#

cudnn_convolution

functionFull reference ↗
tensorplay.functional.cudnn_convolution(input, weight, padding, stride, dilation, groups, benchmark, deterministic, allow_tf32, *, out=None)[source]
#

embedding

functionFull reference ↗
tensorplay.functional.embedding(weight, indices, padding_idx=-1, scale_grad_by_freq=False, sparse=False)[source]
#

empty_like

functionFull reference ↗
tensorplay.functional.empty_like(input, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

empty_permuted

functionFull reference ↗
tensorplay.functional.empty_permuted(size, physical_layout, dtype=None, layout=None, device=None, pin_memory=None)[source]
#

empty_quantized

functionFull reference ↗
tensorplay.functional.empty_quantized(size, qtensor, dtype=None, layout=None, device=None, pin_memory=None, memory_format=None)[source]
#

empty_strided

functionFull reference ↗
tensorplay.functional.empty_strided(size, stride, dtype=None, device=None, pin_memory=False)[source]
#

eye

functionFull reference ↗
tensorplay.functional.eye(n, m=-1, *, dtype=<DType.float32: 8>, device=None, requires_grad=False, out=None)[source]
#

fake_quantize_per_channel_affine

functionFull reference ↗
tensorplay.functional.fake_quantize_per_channel_affine(input, scale, zero_point, axis, quant_min, quant_max)[source]
#

fake_quantize_per_tensor_affine

functionFull reference ↗
tensorplay.functional.fake_quantize_per_tensor_affine(input, scale, zero_point, quant_min, quant_max)[source]
#

fbgemm_linear_fp16_weight_fp32_activation

functionFull reference ↗
tensorplay.functional.fbgemm_linear_fp16_weight_fp32_activation(input, packed_weight, bias, *, out=None)[source]
#

fbgemm_linear_fp16_weight

functionFull reference ↗
tensorplay.functional.fbgemm_linear_fp16_weight(input, packed_weight, bias, *, out=None)[source]
#

fbgemm_linear_int8_weight_fp32_activation

functionFull reference ↗
tensorplay.functional.fbgemm_linear_int8_weight_fp32_activation(input, weight, packed, col_offsets, weight_scale, weight_zero_point, bias)[source]
#

fbgemm_linear_int8_weight

functionFull reference ↗
tensorplay.functional.fbgemm_linear_int8_weight(input, weight, packed, col_offsets, weight_scale, weight_zero_point, bias)[source]
#

fbgemm_pack_quantized_matrix

functionFull reference ↗
tensorplay.functional.fbgemm_pack_quantized_matrix(input, K=None, N=None)[source]
#

from_file

functionFull reference ↗
tensorplay.functional.from_file(filename, shared=None, size=0, dtype=None, layout=None, device=None, pin_memory=None)[source]
#

full_like

functionFull reference ↗
tensorplay.functional.full_like(input, fill_value, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

fused_moving_avg_obs_fake_quant

functionFull reference ↗
tensorplay.functional.fused_moving_avg_obs_fake_quant(input, observer_on, fake_quant_on, running_min, running_max, scale, zero_point, averaging_const, quant_min, quant_max, ch_axis, per_row_fake_quant=False, symmetric_quant=False)[source]
#

grid_sampler_2d

functionFull reference ↗
tensorplay.functional.grid_sampler_2d(input, grid, interpolation_mode, padding_mode, align_corners)[source]
#

grid_sampler_3d

functionFull reference ↗
tensorplay.functional.grid_sampler_3d(input, grid, interpolation_mode, padding_mode, align_corners)[source]
#

gru

functionFull reference ↗
tensorplay.functional.gru(input, hx, params, has_biases=True, num_layers=1, dropout_p=0.0, training=False, bidirectional=False, batch_first=False)[source]
#

hamming_window

functionFull reference ↗
tensorplay.functional.hamming_window(window_length, periodic=True, alpha=0.54, beta=0.46, dtype=None)[source]
#

index_reduce

functionFull reference ↗
tensorplay.functional.index_reduce(input, dim, index, source, reduce, *, include_self=True, out=None)[source]
#

instance_norm

functionFull reference ↗
tensorplay.functional.instance_norm(input, weight=None, bias=None, running_mean=None, running_var=None, use_input_stats=True, momentum=0.1, eps=1e-05)[source]
#

isin

functionFull reference ↗
tensorplay.functional.isin(elements, test_elements, *, assume_unique=False, invert=False, out=None)[source]
#

istft

functionFull reference ↗
tensorplay.functional.istft(input, n_fft, hop_length=None, win_length=None, window=None, center=True, normalized=False, onesided=True, length=None, return_complex=False)[source]
#

kaiser_window

functionFull reference ↗
tensorplay.functional.kaiser_window(window_length, periodic=True, beta=12.0, *, dtype=None, device=None, pin_memory=False)[source]
#

linspace

functionFull reference ↗
tensorplay.functional.linspace(start, end, steps, *, dtype=<DType.float32: 8>, device=None, requires_grad=False, out=None)[source]
#

logspace

functionFull reference ↗
tensorplay.functional.logspace(start, end, steps, base=10.0, *, dtype=<DType.float32: 8>, device=None, requires_grad=False, out=None)[source]
#

lstm

functionFull reference ↗
tensorplay.functional.lstm(input, hx, params, has_biases=True, num_layers=1, dropout_p=0.0, training=False, bidirectional=False, batch_first=False)[source]
#

lu_unpack

functionFull reference ↗
tensorplay.functional.lu_unpack(LU_data, LU_pivots, unpack_data=True, unpack_pivots=True, *, out=None)[source]
#

max_pool1d

functionFull reference ↗
tensorplay.functional.max_pool1d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

max_pool3d

functionFull reference ↗
tensorplay.functional.max_pool3d(input, kernel_size, stride=[], padding=[0, 0, 0], dilation=[1, 1, 1], ceil_mode=False)[source]
#

miopen_batch_norm

functionFull reference ↗
tensorplay.functional.miopen_batch_norm(input, weight, bias, running_mean, running_var, training, exponential_average_factor, epsilon)[source]
#

miopen_convolution_add_relu

functionFull reference ↗
tensorplay.functional.miopen_convolution_add_relu(input, weight, z, alpha, bias, stride, padding, dilation, groups)[source]
#

miopen_convolution_relu

functionFull reference ↗
tensorplay.functional.miopen_convolution_relu(input, weight, bias, stride, padding, dilation, groups)[source]
#

miopen_convolution_transpose

functionFull reference ↗
tensorplay.functional.miopen_convolution_transpose(input, weight, bias, padding, output_padding, stride, dilation, groups, benchmark, deterministic)[source]
#

miopen_convolution

functionFull reference ↗
tensorplay.functional.miopen_convolution(input, weight, bias, padding, stride, dilation, groups, benchmark, deterministic)[source]
#

miopen_ctc_loss

functionFull reference ↗
tensorplay.functional.miopen_ctc_loss(log_probs, targets, input_lengths, target_lengths, blank, deterministic, zero_infinity)[source]
#

miopen_depthwise_convolution

functionFull reference ↗
tensorplay.functional.miopen_depthwise_convolution(input, weight, bias, padding, stride, dilation, groups, benchmark, deterministic)[source]
#

miopen_rnn

functionFull reference ↗
tensorplay.functional.miopen_rnn(input, weight, weight_stride0, hx, cx, mode, hidden_size, num_layers, batch_first, dropout, train, bidirectional, batch_sizes, dropout_state)[source]
#

mkldnn_adaptive_avg_pool2d

functionFull reference ↗
tensorplay.functional.mkldnn_adaptive_avg_pool2d(input, output_size, *, out=None)[source]
#

mkldnn_convolution

functionFull reference ↗
tensorplay.functional.mkldnn_convolution(input, weight, bias, padding, stride, dilation, groups)[source]
#

mkldnn_linear_backward_weights

functionFull reference ↗
tensorplay.functional.mkldnn_linear_backward_weights(grad_output, input, weight, bias_defined)[source]
#

mkldnn_max_pool2d

functionFull reference ↗
tensorplay.functional.mkldnn_max_pool2d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

mkldnn_max_pool3d

functionFull reference ↗
tensorplay.functional.mkldnn_max_pool3d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

mkldnn_rnn_layer

functionFull reference ↗
tensorplay.functional.mkldnn_rnn_layer(input, weight0, weight1, weight2, weight3, hx_, cx_, reverse, batch_sizes, mode, hidden_size, num_layers, has_biases, bidirectional, batch_first, train)[source]
#

multinomial

functionFull reference ↗
tensorplay.functional.multinomial(input, num_samples, replacement=False, impl=0, *, out=None)[source]
#

native_batch_norm

functionFull reference ↗
tensorplay.functional.native_batch_norm(input, weight, bias, running_mean, running_var, training, momentum, eps, *, out=None)[source]
#

native_group_norm

functionFull reference ↗
tensorplay.functional.native_group_norm(input, weight, bias, N, C, HxW, group, eps)[source]
#

native_layer_norm

functionFull reference ↗
tensorplay.functional.native_layer_norm(input, normalized_shape, weight, bias, eps)[source]
#

ones_like

functionFull reference ↗
tensorplay.functional.ones_like(input, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

poisson_nll_loss

functionFull reference ↗
tensorplay.functional.poisson_nll_loss(input, target, log_input=True, full=False, eps=1e-08)[source]
#

quantize_per_channel

functionFull reference ↗
tensorplay.functional.quantize_per_channel(input, scales, zero_points, axis, dtype)[source]
#

quantize_per_tensor_dynamic

functionFull reference ↗
tensorplay.functional.quantize_per_tensor_dynamic(input, dtype, reduce_range)[source]
#

quantized_batch_norm

functionFull reference ↗
tensorplay.functional.quantized_batch_norm(input, weight, bias, mean, var, eps, output_scale, output_zero_point)[source]
#

quantized_gru_cell

functionFull reference ↗
tensorplay.functional.quantized_gru_cell(input, hx, w_ih, w_hh, b_ih, b_hh, packed_ih, packed_hh, col_offsets_ih, col_offsets_hh, scale_ih, scale_hh, zero_point_ih, zero_point_hh)[source]
#

quantized_lstm_cell

functionFull reference ↗
tensorplay.functional.quantized_lstm_cell(input, hx, w_ih, w_hh, b_ih, b_hh, packed_ih, packed_hh, col_offsets_ih, col_offsets_hh, scale_ih, scale_hh, zero_point_ih, zero_point_hh)[source]
#

quantized_max_pool1d

functionFull reference ↗
tensorplay.functional.quantized_max_pool1d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

quantized_max_pool2d

functionFull reference ↗
tensorplay.functional.quantized_max_pool2d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

quantized_max_pool3d

functionFull reference ↗
tensorplay.functional.quantized_max_pool3d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False)[source]
#

quantized_rnn_relu_cell

functionFull reference ↗
tensorplay.functional.quantized_rnn_relu_cell(input, hx, w_ih, w_hh, b_ih, b_hh, packed_ih, packed_hh, col_offsets_ih, col_offsets_hh, scale_ih, scale_hh, zero_point_ih, zero_point_hh)[source]
#

quantized_rnn_tanh_cell

functionFull reference ↗
tensorplay.functional.quantized_rnn_tanh_cell(input, hx, w_ih, w_hh, b_ih, b_hh, packed_ih, packed_hh, col_offsets_ih, col_offsets_hh, scale_ih, scale_hh, zero_point_ih, zero_point_hh)[source]
#

rand_like

functionFull reference ↗
tensorplay.functional.rand_like(input, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

randint_like

functionFull reference ↗
tensorplay.functional.randint_like(input, low, high, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

randint

functionFull reference ↗
tensorplay.functional.randint(low, high, size, *, dtype=<DType.int64: 4>, device=None, requires_grad=False, out=None)[source]
#

randn_like

functionFull reference ↗
tensorplay.functional.randn_like(input, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]
#

randperm

functionFull reference ↗
tensorplay.functional.randperm(n, *, dtype=<DType.int64: 4>, device=None, requires_grad=False, out=None)[source]
#

repeat_interleave

functionFull reference ↗
tensorplay.functional.repeat_interleave(input, repeats=None, dim=None, *, output_size=None)[source]
#

rnn_relu

functionFull reference ↗
tensorplay.functional.rnn_relu(input, hx, params, has_biases=True, num_layers=1, dropout_p=0.0, training=False, bidirectional=False, batch_first=False)[source]
#

rnn_tanh

functionFull reference ↗
tensorplay.functional.rnn_tanh(input, hx, params, has_biases=True, num_layers=1, dropout_p=0.0, training=False, bidirectional=False, batch_first=False)[source]
#

scatter_reduce

functionFull reference ↗
tensorplay.functional.scatter_reduce(input, dim, index, src, reduce, *, include_self=True, out=None)[source]
#

searchsorted

functionFull reference ↗
tensorplay.functional.searchsorted(sorted_sequence, input, *, out_int32=False, right=False, side=None, sorter=None, out=None)[source]
#

segment_reduce

functionFull reference ↗
tensorplay.functional.segment_reduce(data, reduce, lengths=None, indices=None, offsets=None, axis=0, unsafe=False, initial=None)[source]
#

sort

functionFull reference ↗
tensorplay.functional.sort(input, dim=-1, descending=False, *, stable=<object object>, out=None)[source]
#

sparse_bsc_tensor

functionFull reference ↗
tensorplay.functional.sparse_bsc_tensor(ccol_indices, row_indices, values, size=None, *, dtype=None, layout=None, device=None, pin_memory=False)[source]
#

sparse_bsr_tensor

functionFull reference ↗
tensorplay.functional.sparse_bsr_tensor(crow_indices, col_indices, values, size=None, *, dtype=None, layout=None, device=None, pin_memory=False)[source]
#

sparse_coo_tensor

functionFull reference ↗
tensorplay.functional.sparse_coo_tensor(indices, values, size=None, is_coalesced=False)[source]
#

sparse_csc_tensor

functionFull reference ↗
tensorplay.functional.sparse_csc_tensor(ccol_indices, row_indices, values, size=None, *, dtype=None, layout=None, device=None, pin_memory=False)[source]
#

sparse_csr_tensor

functionFull reference ↗
tensorplay.functional.sparse_csr_tensor(crow_indices, col_indices, values, size=None, *, dtype=None, layout=None, device=None, pin_memory=False)[source]
#

split_with_sizes_copy

functionFull reference ↗
tensorplay.functional.split_with_sizes_copy(input, split_sizes, dim=0, *, out=None)[source]
#

stft

functionFull reference ↗
tensorplay.functional.stft(input, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True, return_complex=True)[source]
#

sym_constrain_range_for_size

functionFull reference ↗
tensorplay.functional.sym_constrain_range_for_size(size, min=None, max=None)[source]
#

triangular_solve

functionFull reference ↗
tensorplay.functional.triangular_solve(input, A, upper=False, transpose=False, unitriangular=False, *, out=None)[source]
#

tril_indices

functionFull reference ↗
tensorplay.functional.tril_indices(row, col, offset=0, dtype=<DType.int64: 4>, device=None, pin_memory=False)[source]
#

triplet_margin_loss

functionFull reference ↗
tensorplay.functional.triplet_margin_loss(anchor, positive, negative, margin=1.0, p=2.0)[source]
#

triu_indices

functionFull reference ↗
tensorplay.functional.triu_indices(row, col, offset=0, dtype=<DType.int64: 4>, device=None, pin_memory=False)[source]
#

unique_consecutive

functionFull reference ↗
tensorplay.functional.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)[source]
#

unique

functionFull reference ↗
tensorplay.functional.unique(input, sorted=True, return_inverse=False, return_counts=False, dim=None)[source]
#

zeros_like

functionFull reference ↗
tensorplay.functional.zeros_like(input, dtype=<DType.undefined: 22>, device=None, requires_grad=False)[source]

Search documentation

Search all 1,743 documentation pages.

Keyboard shortcuts

Global

  • /Focus search
  • ?This dialog
  • ,Open settings
  • jAI assistant

Search

  • Navigate results
  • Open result
  • escClose

Package

  • mMain information
  • dDocs
  • .Code
  • -Changelog
  • tTimeline
  • sStats
  • vVersions