Torch.Functional.Internal
align_tensors Source #
Arguments
tensors
native_dropout Source #
input
p
train
dropout Source #
feature_dropout Source #
alpha_dropout Source #
feature_alpha_dropout Source #
abs Source #
self
absolute Source #
angle Source #
view_as_real Source #
view_as_complex Source #
sgn Source #
real Source #
imag Source #
conj Source #
conj_physical Source #
resolve_conj Source #
resolve_neg Source #
acos Source #
arccos Source #
avg_pool1d Source #
kernel_size
stride
padding
ceil_mode
count_include_pad
adaptive_avg_pool1d Source #
output_size
adaptive_max_pool1d Source #
add Source #
other
alpha
addScalar Source #
addmv Source #
mat
vec
beta
addr Source #
vec1
vec2
affine_grid_generator Source #
theta
size
align_corners
allDim Source #
dim
keepdim
allWithDimname Source #
allclose Source #
rtol
atol
equal_nan
anyDim Source #
anyWithDimname Source #
argmax Source #
argmin Source #
acosh Source #
arccosh Source #
asinh Source #
arcsinh Source #
atanh Source #
arctanh Source #
as_strided Source #
storage_offset
asin Source #
arcsin Source #
atan Source #
arctan Source #
atleast_1d_t Source #
atleast_1d_l Source #
atleast_2d_t Source #
atleast_2d_l Source #
atleast_3d_t Source #
atleast_3d_l Source #
baddbmm Source #
batch1
batch2
batch_norm Source #
weight
bias
running_mean
running_var
training
momentum
eps
cudnn_enabled
quantized_batch_norm Source #
mean
var
output_scale
output_zero_point
bilinear Source #
input1
input2
binary_cross_entropy Source #
target
reduction
binary_cross_entropy_with_logits Source #
pos_weight
bincount Source #
weights
minlength
bitwise_not Source #
copysign_tt Source #
copysign_ts Source #
logical_not Source #
logical_xor Source #
logical_and Source #
logical_or Source #
bmm Source #
mat2
broadcast_tensors Source #
broadcast_to Source #
cat Source #
catWithDimname Source #
concat_ll Source #
concat_ln Source #
block_diag Source #
ceil Source #
chain_matmul Source #
matrices
unsafe_chunk Source #
chunks
chunk Source #
tensor_split_tll Source #
sections
clamp_tss Source #
min
max
clamp_ttt Source #
clamp_max_ts Source #
clamp_max_tt Source #
clamp_min_ts Source #
clamp_min_tt Source #
clip_tss Source #
clip_ttt Source #
cudnn_is_acceptable Source #
complex Source #
real
imag
polar Source #
abs
angle
constant_pad_nd Source #
pad
value
convolution Source #
dilation
transposed
output_padding
groups
convolution_overrideable Source #
convolution_backward_overrideable Source #
grad_output
output_mask
conv1d_tttllll Source #
conv2d Source #
conv3d_tttllll Source #
conv1d_tttlsll Source #
conv2d_tttlsll Source #
conv3d_tttlsll Source #
conv_tbc Source #
conv_transpose1d Source #
conv_transpose2d Source #
conv_transpose3d Source #
cos Source #
cosh Source #
cosine_embedding_loss Source #
margin
count_nonzero_tl Source #
cov Source #
correction
fweights
aweights
corrcoef Source #
cudnn_affine_grid_generator Source #
N
C
H
W
cudnn_batch_norm Source #
exponential_average_factor
epsilon
cudnn_convolution Source #
benchmark
deterministic
allow_tf32
cudnn_convolution_transpose Source #
cudnn_convolution_relu Source #
cudnn_convolution_add_relu Source #
z
cudnn_grid_sampler Source #
grid
cummax_tl Source #
cummax_tn Source #
cummin_tl Source #
cummin_tn Source #
cumprod Source #
dtype
cumprodWithDimname Source #
cumsum Source #
cumsumWithDimname Source #
cumulative_trapezoid_ttl Source #
y
x
cumulative_trapezoid_tsl Source #
dx
ctcLoss' Source #
log_probs
targets
input_lengths
target_lengths
blank
zero_infinity
ctcLoss Source #
diag_embed Source #
offset
dim1
dim2
diagflat Source #
diagonal_tlll Source #
linalg_diagonal Source #
A
diagonal_tnnnl Source #
outdim
diff Source #
n
prepend
append
gradient_tsll Source #
spacing
edge_order
gradient_tll Source #
div Source #
div_tts Source #
rounding_mode
divScalar Source #
div_tss Source #
divide_tt Source #
divide_ts Source #
divide_tts Source #
divide_tss Source #
true_divide_tt Source #
true_divide_ts Source #
dot Source #
tensor
vdot Source #
einsum Source #
equation
embedding Source #
indices
padding_idx
scale_grad_by_freq
sparse
row_stack Source #
embedding_bag_tttblbtb Source #
offsets
mode
per_sample_weights
include_last_offset
embedding_bag_tttblbtbl Source #
erf Source #
erfc Source #
exp Source #
exp2 Source #
expm1 Source #
flatten Source #
start_dim
end_dim
flattenTo Source #
out_dim
flattenToWithDimname Source #
flattenToWithDimnames Source #
dims
floor Source #
floor_divide_tt Source #
floor_divide_ts Source #
frac Source #
gcd Source #
lcm Source #
grid_sampler Source #
interpolation_mode
padding_mode
grid_sampler_2d Source #
grid_sampler_3d Source #
hinge_embedding_loss Source #
group_norm Source #
num_groups
native_group_norm Source #
HxW
group
index Source #
indexCopy Source #
index
source
indexCopyWithDimname Source #
index_put Source #
values
accumulate
instance_norm Source #
use_input_stats
inverse Source #
isclose Source #
isin_ttbb Source #
elements
test_elements
assume_unique
invert
isin_tsbb Source #
test_element
isin_stbb Source #
element
isnan Source #
is_distributed Source #
is_floating_point Source #
is_complex Source #
is_conj Source #
is_neg Source #
isreal Source #
is_nonzero Source #
is_same_size Source #
is_signed Source #
is_inference Source #
kl_div Source #
log_target
kron Source #
kthvalue Source #
k
kthvalueWithDimname Source #
layer_norm Source #
normalized_shape
cudnn_enable
native_layer_norm Source #
nan_to_num Source #
nan
posinf
neginf
linear Source #
mkldnn_linear Source #
mkldnn_linear_backward_input Source #
input_size
mkldnn_linear_backward_weights Source #
bias_defined
fbgemm_linear_int8_weight_fp32_activation Source #
packed
col_offsets
weight_scale
weight_zero_point
fbgemm_linear_int8_weight Source #
fbgemm_linear_quantize_weight Source #
fbgemm_pack_gemm_matrix_fp16 Source #
fbgemm_linear_fp16_weight_fp32_activation Source #
packed_weight
fbgemm_linear_fp16_weight Source #
quantizeFbgemm' Source #
quantizeFbgemm Source #
K
ldexp Source #
log Source #
log10 Source #
log1p Source #
log2 Source #
logaddexp Source #
logaddexp2 Source #
xlogy_tt Source #
xlogy_st Source #
xlogy_ts Source #
logdet Source #
logSoftmax Source #
logSoftmaxWithDimname Source #
logcumsumexp_tl Source #
logcumsumexp_tn Source #
logsumexp Source #
logsumexpWithDimnameList Source #
margin_ranking_loss Source #
matmul Source #
matrixRank Source #
tol
symmetric
matrixRank' Source #
matrix_power Source #
matrix_exp Source #
aminmax Source #
maxDim Source #
maxWithDimname Source #
amax Source #
max_pool1d_with_indices Source #
max_pool1d Source #
max_pool2d Source #
mkldnn_max_pool2d Source #
mkldnn_max_pool3d Source #
quantized_max_pool1d Source #
quantized_max_pool2d Source #
max_pool3d Source #
meanAll Source #
meanDim Source #
meanWithDimnames Source #
nanmean Source #
medianAll Source #
medianDim Source #
medianWithDimname Source #
nanmedian_t Source #
nanmedian_tlb Source #
nanmedian_tnb Source #
minDim Source #
minWithDimname Source #
amin Source #
mkldnn_convolution Source #
miopen_batch_norm Source #
miopen_convolution Source #
miopen_convolution_transpose Source #
miopen_depthwise_convolution Source #
miopen_rnn Source #
weight_stride0
hx
cx
hidden_size
num_layers
batch_first
dropout
bidirectional
batch_sizes
dropout_state
mm Source #
mode Source #
modeWithDimname Source #
mul Source #
mulScalar Source #
multiply_tt Source #
multiply_ts Source #
mv Source #
mvlgamma Source #
narrow_copy Source #
start
length
narrow_tlll Source #
narrow_tltl Source #
native_batch_norm Source #
batch_norm_stats Source #
batch_norm_elemt Source #
invstd
batch_norm_gather_stats Source #
count
batch_norm_gather_stats_with_counts Source #
counts
batch_norm_backward_reduce Source #
grad_out
input_g
weight_g
bias_g
batch_norm_backward_elemt Source #
mean_dy
mean_dy_xmu
batch_norm_update_stats Source #
is_vulkan_available :: Bool Source #
pairwise_distance Source #
x1
x2
cdist Source #
compute_mode
pdist Source #
cosine_similarity Source #
permute Source #
movedim_tll Source #
destination
moveaxis_tll Source #
adjoint Source #
pixel_shuffle Source #
upscale_factor
pixel_unshuffle Source #
downscale_factor
channel_shuffle Source #
native_channel_shuffle Source #
pinverse Source #
rcond
poisson_nll_loss Source #
log_input
full
rad2deg Source #
deg2rad Source #
ravel Source #
reciprocal Source #
neg Source #
negative Source #
repeat_interleave_tl Source #
repeats
repeat_interleave_ttll Source #
repeat_interleave_tlll Source #
reshape Source #
shape
round_t Source #
round_tl Source #
decimals
relu Source #
relu6 Source #
prelu Source #
gelu Source #
hardshrink Source #
lambd
rsqrt Source #
selectWithDimname Source #
select Source #
selu Source #
celu Source #
silu Source #
mish Source #
sigmoid Source #
logit Source #
sin Source #
sinc Source #
sinh Source #
size Source #
sizeWithDimname Source #
slice Source #
end
step
slice_scatter Source #
src
select_scatter Source #
diagonal_scatter Source #
slogdet Source #
smm Source #
softmax Source #
softmaxWithDimname Source #
unsafe_split Source #
split_size
split Source #
unsafe_split_with_sizes Source #
split_sizes
split_with_sizes Source #
hsplit_tl Source #
vsplit_tl Source #
dsplit_tl Source #
squeezeAll Source #
squeezeDim Source #
squeezeWithDimname Source #
sspaddmm Source #
mat1
stack Source #
hstack Source #
vstack Source #
dstack Source #
stft Source #
n_fft
hop_length
win_length
window
normalized
onesided
return_complex
istft Source #
center
stride Source #
strideWithDimname Source #
sumAll Source #
sumDim Source #
sumWithDimnames Source #
nansum_ts Source #
nansum_tlbs Source #
sqrt Source #
square Source #
stdAll Source #
unbiased
stdDim Source #
std_tllb Source #
stdMeanAll Source #
stdMeanDim Source #
std_mean_tllb Source #
stdMeanWithDimnames Source #
std_mean_tNlb Source #
stdWithDimnames Source #
std_tNlb Source #
prodAll Source #
prodDim Source #
prodWithDimnames Source #
t Source #
tan Source #
tanh Source #
tensordot Source #
dims_self
dims_other
threshold Source #
threshold
tile Source #
transpose Source #
dim0
transposeWithDimname Source #
one_hot Source #
num_classes
flip Source #
fliplr Source #
flipud Source #
roll Source #
shifts
rot90 Source #
trapezoid_ttl Source #
trapezoid_tsl Source #
trapz Source #
trapzScalar Source #
triplet_margin_loss Source #
anchor
positive
negative
swap
trunc Source #
fix Source #
unique_dim Source #
sorted
return_inverse
return_counts
unique_consecutive Source #
unique_dim_consecutive Source #
unsqueeze Source #
vander Source #
increasing
var Source #
varDim Source #
var_tllb Source #
varWithDimnames Source #
var_tNlb Source #
varMean Source #
varMeanDim Source #
var_mean_tllb Source #
varMeanWithDimnames Source #
var_mean_tNlb Source #
where' Source #
condition
where_tst Source #
where_tts Source #
where_tss Source #
isNonZero Source #
norm_except_dim Source #
v
pow
native_norm_ts Source #
native_norm_tslbs Source #
normCastAll Source #
normAll Source #
normCastDim Source #
normDim Source #
norm_tsNbs Source #
norm_tsNb Source #
frexp Source #
frobeniusNormAll Source #
frobeniusNormDim Source #
nuclearNormAll Source #
nuclearNormDim Source #
clone Source #
memory_format
positive Source #
sub Source #
subScalar Source #
subtract_tts Source #
subtract_tss Source #
rsub Source #
heaviside Source #
rsubScalar Source #
sparse_sampled_addmm Source #
addmm Source #
hspmm Source #
unbind Source #
unbindWithDimname Source #
mkldnn_reorder_conv2d_weight Source #
mkldnn_reorder_conv3d_weight Source #
quantize_per_tensor_dynamic Source #
reduce_range
quantize_per_tensor_tdls Source #
scale
zero_point
quantize_per_tensor_ttts Source #
quantize_per_tensor_ltts Source #
scales
zero_points
quantize_per_channel Source #
axis
dequantize_t Source #
dequantize_l Source #
q_scale Source #
q_zero_point Source #
q_per_channel_scales Source #
q_per_channel_zero_points Source #
q_per_channel_axis Source #
int_repr Source #
fake_quantize_per_tensor_affine_tdlll Source #
quant_min
quant_max
fake_quantize_per_tensor_affine_tttll Source #
fake_quantize_per_tensor_affine_cachemask Source #
fake_quantize_per_channel_affine Source #
fake_quantize_per_channel_affine_cachemask Source #
fused_moving_avg_obs_fake_quant Source #
observer_on
fake_quant_on
running_min
running_max
averaging_const
ch_axis
per_row_fake_quant
symmetric_quant
choose_qparams_optimized Source #
numel
n_bins
ratio
bit_width
meshgrid_l Source #
meshgrid_ls Source #
indexing
cartesian_prod Source #
combinations Source #
r
with_replacement
resultType Source #
resultTypeScalar Source #
resultTypeScalar' Source #
scalar
resultTypeScalars Source #
scalar1
scalar2
can_cast Source #
from
to
promote_types Source #
type1
type2
lstm Source #
params
has_biases
lstm' Source #
data
gru Source #
gru' Source #
rnnTanh Source #
rnnTanh' Source #
rnnRelu Source #
rnnRelu' Source #
lstm_cell Source #
w_ih
w_hh
b_ih
b_hh
gru_cell Source #
rnn_tanh_cell Source #
rnn_relu_cell Source #
quantized_lstm_cell Source #
packed_ih
packed_hh
col_offsets_ih
col_offsets_hh
scale_ih
scale_hh
zero_point_ih
zero_point_hh
quantized_gru_cell Source #
quantized_rnn_relu_cell Source #
quantized_rnn_tanh_cell Source #
maskedFillScalar Source #
mask
maskedFill Source #
masked_scatter Source #
put Source #
index_add_tltts Source #
index_add_tntts Source #
indexFillScalar Source #
indexFill Source #
indexFillScalarWithDimname Source #
indexFillWithDimname Source #
scatter Source #
scatterScalar Source #
scatter_tltts Source #
reduce
scatter_tltss Source #
scatterWithDimname Source #
scatterScalarWithDimname Source #
scatterAdd Source #
scatterAddWithDimname Source #
scatter_reduce Source #
bitwise_and_ts Source #
bitwise_and_tt Source #
bitwise_or_ts Source #
bitwise_or_tt Source #
bitwiseXorScalar Source #
bitwiseXor Source #
bitwise_left_shift_tt Source #
bitwise_left_shift_ts Source #
bitwise_left_shift_st Source #
bitwise_right_shift_tt Source #
bitwise_right_shift_ts Source #
bitwise_right_shift_st Source #
addbmm Source #
diag Source #
diagonal
cross Source #
triu Source #
tril Source #
trace Source #
neScalar Source #
ne Source #
not_equal_ts Source #
not_equal_tt Source #
eqScalar Source #
eq Source #
geScalar Source #
ge Source #
greater_equal_ts Source #
greater_equal_tt Source #
leScalar Source #
le Source #
less_equal_ts Source #
less_equal_tt Source #
gtScalar Source #
gt Source #
greater_ts Source #
greater_tt Source #
ltScalar Source #
lt Source #
less_ts Source #
less_tt Source #
take Source #
take_along_dim Source #
indexSelect Source #
indexSelectWithDimname Source #
masked_select Source #
nonzero Source #
nonzero_numpy Source #
argwhere Source #
gather Source #
sparse_grad
gatherWithDimname Source #
addcmul Source #
tensor1
tensor2
addcdiv Source #
cross_entropy_loss Source #
ignore_index
label_smoothing
lstsq Source #
triangular_solve Source #
upper
transpose
unitriangular
linalg_solve_triangular Source #
B
left
symeig Source #
eigenvectors
eig Source #
svd Source #
some
compute_uv
swapaxes Source #
axis0
axis1
swapdims Source #
cholesky Source #
cholesky_solve Source #
solve Source #
cholesky_inverse Source #
qr Source #
geqrf Source #
orgqr Source #
ormqr Source #
input3
lu_solve Source #
LU_data
LU_pivots
lu_unpack Source #
unpack_data
unpack_pivots
lgamma Source #
digamma Source #
polygamma Source #
erfinv Source #
i0 Source #
sign Source #
signbit Source #
dist Source #
atan2 Source #
arctan2 Source #
lerpScalar Source #
lerp Source #
histc Source #
bins
histogram_tttb Source #
density
histogram_tlatb Source #
range
fmodScalar Source #
fmod Source #
hypot Source #
igamma Source #
igammac Source #
nextafter Source #
remainderScalar Source #
remainder Source #
remainder_st Source #
minAll Source #
fmin Source #
maxAll Source #
fmax Source #
maximum Source #
max Source #
minimum Source #
min Source #
quantile_ttlbs Source #
q
interpolation
quantile_tdlbs Source #
nanquantile_ttlbs Source #
nanquantile_tdlbs Source #
sort Source #
descending
sort_tblb Source #
stable
sortWithDimname Source #
sort_tbnb Source #
msort Source #
argsort Source #
argsortWithDimname Source #
topk Source #
largest
all Source #
any Source #
renorm Source #
maxnorm
equal Source #
pow Source #
exponent
powScalar' Source #
powScalar Source #
float_power_tt Source #
float_power_st Source #
float_power_ts Source #
alias Source #
bucketize_ttbb Source #
boundaries
out_int32
right
bucketize_stbb Source #
searchsorted_ttbbst Source #
sorted_sequence
side
sorter
searchsorted_tsbbst Source #
mse_loss Source #
l1_loss Source #
multi_margin_loss Source #
multilabel_margin_loss Source #
multilabel_margin_loss_forward Source #
nll_loss_nd Source #
nll_loss Source #
nll_loss_forward Source #
nll_loss2d Source #
nll_loss2d_forward Source #
smooth_l1_loss Source #
huber_loss Source #
delta
soft_margin_loss Source #
elu Source #
input_scale
glu Source #
hardsigmoid Source #
hardtanh Source #
min_val
max_val
hardswish Source #
leaky_relu Source #
negative_slope
log_sigmoid Source #
log_sigmoid_forward Source #
softplus Source #
softshrink Source #
adaptive_avg_pool2d Source #
mkldnn_adaptive_avg_pool2d Source #
adaptive_avg_pool3d Source #
adaptive_max_pool2d Source #
adaptive_max_pool3d Source #
avg_pool2d Source #
divisor_override
avg_pool3d Source #
fractional_max_pool2d Source #
random_samples
fractional_max_pool3d Source #
max_pool2d_with_indices Source #
max_pool3d_with_indices Source #
max_unpool2d Source #
max_unpool3d Source #
reflection_pad1d Source #
reflection_pad2d Source #
reflection_pad3d Source #
replication_pad1d Source #
replication_pad2d Source #
replication_pad3d Source #
upsample_linear1d_tlba Source #
scale_factors
upsample_bilinear2d_tlba Source #
upsample_trilinear3d_tlba Source #
upsample_bicubic2d_tlba Source #
upsample_nearest1d_tla Source #
upsample_nearest2d_tla Source #
upsample_nearest3d_tla Source #
upsample_bilinear2d_tlbdd Source #
scales_h
scales_w
upsample_bicubic2d_tlbdd Source #
upsample_trilinear3d_tlbddd Source #
scales_d
upsample_nearest2d_tldd Source #
upsample_nearest3d_tlddd Source #
slow_conv_transpose2d Source #
slow_conv_transpose3d Source #
thnn_conv2d Source #
conv_depthwise3d Source #
slow_conv3d Source #
slow_conv3d_forward Source #
slow_conv_dilated2d Source #
slow_conv_dilated3d Source #
col2im Source #
column_stack Source #
im2col Source #
isfinite Source #
isinf Source #
isposinf Source #
isneginf Source #
special_entr Source #
special_ndtri Source #
special_expm1 Source #
special_exp2 Source #
special_psi Source #
special_digamma Source #
special_gammaln Source #
special_erf Source #
special_erfc Source #
special_erfcx Source #
special_erfinv Source #
special_ndtr Source #
special_xlog1py_tt Source #
special_xlog1py_st Source #
special_xlog1py_ts Source #
special_xlogy_tt Source #
special_xlogy_st Source #
special_xlogy_ts Source #
special_zeta_tt Source #
special_zeta_st Source #
special_zeta_ts Source #
special_i0 Source #
special_i0e Source #
special_i1 Source #
special_i1e Source #
special_logit Source #
special_polygamma Source #
special_logsumexp Source #
special_expit Source #
special_sinc Source #
special_round Source #
special_log1p Source #
special_log_softmax Source #
special_gammainc Source #
special_gammaincc Source #
special_multigammaln Source #
special_softmax Source #
fft_fft Source #
norm
fft_ifft Source #
fft_rfft Source #
fft_irfft Source #
fft_hfft Source #
fft_ihfft Source #
fft_fft2 Source #
s
fft_ifft2 Source #
fft_rfft2 Source #
fft_irfft2 Source #
fft_hfft2 Source #
fft_ihfft2 Source #
fft_fftn Source #
fft_ifftn Source #
fft_rfftn Source #
fft_irfftn Source #
fft_hfftn Source #
fft_ihfftn Source #
fft_fftshift Source #
fft_ifftshift Source #
linalg_cholesky_ex Source #
check_errors
linalg_cholesky Source #
linalg_cross Source #
linalg_lu_factor Source #
pivot
linalg_lu_factor_ex Source #
linalg_det Source #
det Source #
linalg_lstsq Source #
b
driver
linalg_matmul Source #
linalg_matrix_exp Source #
linalg_slogdet Source #
linalg_eig Source #
linalg_eigvals Source #
linalg_eigh Source #
UPLO
linalg_eigvalsh Source #
linalg_householder_product Source #
tau
linalg_inv_ex Source #
linalg_inv Source #
inner Source #
outer Source #
ger Source #
linalg_norm_tslbs Source #
ord
linalg_vector_norm Source #
linalg_matrix_norm_tslbs Source #
linalg_svd Source #
full_matrices
linalg_svdvals Source #
linalg_cond_ts Source #
linalg_pinv_tttb Source #
hermitian
linalg_pinv_tddb Source #
linalg_pinv_tdb Source #
linalg_pinv_ttb Source #
linalg_solve Source #
linalg_tensorinv Source #
ind
linalg_tensorsolve Source #
linalg_qr Source #
linalg_matrix_power Source #
linalg_matrix_rank_tttb Source #
linalg_matrix_rank_tddb Source #
linalg_matrix_rank_tdb Source #
linalg_matrix_rank_ttb Source #
linalg_multi_dot Source #
segment_reduce Source #
lengths
unsafe
initial
pad_sequence Source #
sequences
padding_value
flatten_dense_tensors Source #
unflatten_dense_tensors Source #
flat