mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
68f74dd162
Add python and C++ support for LPPool3d to Fixes #114114 Pull Request resolved: https://github.com/pytorch/pytorch/pull/114199 Approved by: https://github.com/mikaylagawarecki
624 lines
7.3 KiB
Plaintext
624 lines
7.3 KiB
Plaintext
__imul__
|
|
__matmul__
|
|
__radd__
|
|
__rand__
|
|
__rdiv__
|
|
__rmatmul__
|
|
__rmod__
|
|
__rmul__
|
|
__ror__
|
|
__rpow__
|
|
__rrshift__
|
|
__rsub__
|
|
__rxor__
|
|
add
|
|
atan2
|
|
bitwise_and
|
|
bitwise_left_shift
|
|
bitwise_or
|
|
bitwise_right_shift
|
|
bitwise_xor
|
|
complex
|
|
copysign
|
|
div
|
|
eq
|
|
float_power
|
|
floor_divide
|
|
fmax
|
|
fmin
|
|
fmod
|
|
gcd
|
|
ge
|
|
gt
|
|
heaviside
|
|
hypot
|
|
igamma
|
|
igammac
|
|
isclose
|
|
lcm
|
|
ldexp
|
|
le
|
|
logical_and
|
|
logical_or
|
|
logical_xor
|
|
lt
|
|
maximum
|
|
minimum
|
|
mul
|
|
ne
|
|
nextafter
|
|
polar
|
|
pow
|
|
remainder
|
|
rsub
|
|
special.xlog1py
|
|
special.zeta
|
|
sub
|
|
true_divide
|
|
xlogy
|
|
abs
|
|
acos
|
|
acosh
|
|
angle
|
|
asin
|
|
asinh
|
|
atan
|
|
atanh
|
|
bitwise_not
|
|
ceil
|
|
conj
|
|
conj_physical
|
|
cos
|
|
cosh
|
|
deg2rad
|
|
digamma
|
|
erf
|
|
erfc
|
|
erfinv
|
|
exp
|
|
exp2
|
|
expm1
|
|
floor
|
|
frac
|
|
frexp
|
|
i0
|
|
imag
|
|
isfinite
|
|
isinf
|
|
isnan
|
|
isneginf
|
|
isposinf
|
|
isreal
|
|
lgamma
|
|
log
|
|
log10
|
|
log1p
|
|
log2
|
|
logical_not
|
|
logit
|
|
multigammaln
|
|
nan_to_num
|
|
neg
|
|
nn.functional.celu
|
|
nn.functional.elu
|
|
nn.functional.hardsigmoid
|
|
nn.functional.logsigmoid
|
|
nn.functional.mish
|
|
nn.functional.rrelu
|
|
nn.functional.selu
|
|
nn.functional.silu
|
|
nn.functional.softsign
|
|
nn.functional.tanhshrink
|
|
polygamma
|
|
positive
|
|
rad2deg
|
|
real
|
|
reciprocal
|
|
round
|
|
rsqrt
|
|
sgn
|
|
sigmoid
|
|
sign
|
|
signbit
|
|
sin
|
|
sinc
|
|
sinh
|
|
special.entr
|
|
special.erfcx
|
|
special.i0e
|
|
special.i1
|
|
special.i1e
|
|
special.log_ndtr
|
|
special.ndtr
|
|
special.ndtri
|
|
special.polygamma
|
|
sqrt
|
|
square
|
|
tan
|
|
tanh
|
|
trunc
|
|
fft_fftfreq
|
|
fft_rfftfreq
|
|
fft.fft
|
|
fft.fft2
|
|
fft.fftn
|
|
fft.fftshift
|
|
fft.hfft
|
|
fft.hfft2
|
|
fft.hfftn
|
|
fft.ifft
|
|
fft.ifft2
|
|
fft.ifftn
|
|
fft.ifftshift
|
|
fft.ihfft
|
|
fft.ihfft2
|
|
fft.ihfftn
|
|
fft.irfft
|
|
fft.irfft2
|
|
fft.irfftn
|
|
fft.rfft
|
|
fft.rfft2
|
|
fft.rfftn
|
|
istft
|
|
stft
|
|
as_strided
|
|
atleast_1d
|
|
atleast_2d
|
|
atleast_3d
|
|
broadcast_shapes
|
|
broadcast_tensors
|
|
broadcast_to
|
|
cat
|
|
chunk
|
|
column_stack
|
|
contiguous
|
|
dsplit
|
|
dstack
|
|
expand
|
|
expand_as
|
|
flatten
|
|
flip
|
|
fliplr
|
|
flipud
|
|
H
|
|
hsplit
|
|
hstack
|
|
mH / adjoint
|
|
movedim
|
|
mT
|
|
narrow
|
|
narrow_copy
|
|
permute
|
|
ravel
|
|
repeat
|
|
repeat_interleave
|
|
reshape
|
|
reshape_as
|
|
resize_
|
|
resize_as_
|
|
roll
|
|
rot90
|
|
split
|
|
split_with_sizes
|
|
squeeze
|
|
stack
|
|
swapaxes
|
|
T
|
|
t
|
|
tensor_split
|
|
tile
|
|
transpose
|
|
unbind
|
|
unflatten
|
|
unfold
|
|
unsqueeze
|
|
view
|
|
view_as
|
|
vsplit
|
|
vstack
|
|
addbmm
|
|
addmm
|
|
addmv
|
|
addr
|
|
baddbmm
|
|
block_diag
|
|
bmm
|
|
cartesian_prod
|
|
chain_matmul
|
|
cholesky
|
|
cholesky_inverse
|
|
cholesky_solve
|
|
cross
|
|
diag
|
|
diag_embed
|
|
diagflat
|
|
diagonal
|
|
dot
|
|
eig
|
|
einsum
|
|
frobenius_norm
|
|
geqrf
|
|
ger
|
|
inner
|
|
inverse
|
|
kron
|
|
linalg.cholesky
|
|
linalg.cholesky_ex
|
|
linalg.cond
|
|
linalg.cross
|
|
linalg.det
|
|
linalg.eig
|
|
linalg.eigh
|
|
linalg.eigvals
|
|
linalg.eigvalsh
|
|
linalg.householder_product
|
|
linalg.inv
|
|
linalg.inv_ex
|
|
linalg.lstsq
|
|
linalg.lu_factor
|
|
linalg.lu_factor_ex
|
|
linalg.matrix_norm
|
|
linalg.matrix_power
|
|
linalg.matrix_rank
|
|
linalg.multi_dot
|
|
linalg.norm
|
|
linalg.pinv
|
|
linalg.qr
|
|
linalg.slogdet
|
|
linalg.solve
|
|
linalg.solve_triangular
|
|
linalg.svd
|
|
linalg.svdvals
|
|
linalg.tensorinv
|
|
linalg.tensorsolve
|
|
linalg.vector_norm
|
|
lobpcg
|
|
logaddexp
|
|
logaddexp2
|
|
logdet
|
|
lu
|
|
lu_solve
|
|
lu_unpack
|
|
matmul
|
|
matrix_exp
|
|
mm
|
|
mv
|
|
norm
|
|
nuclear_norm
|
|
orgqr
|
|
ormqr
|
|
outer
|
|
pca_lowrank
|
|
pinverse
|
|
qr
|
|
renorm
|
|
solve
|
|
svd
|
|
svd_lowrank
|
|
symeig
|
|
tensordot
|
|
trace
|
|
triangular_solve
|
|
tril
|
|
tril_indices
|
|
triu
|
|
triu_indices
|
|
vdot
|
|
all
|
|
amax
|
|
amin
|
|
nanquantile
|
|
quantile
|
|
sum
|
|
nansum
|
|
aminmax
|
|
any
|
|
argmax
|
|
std
|
|
var
|
|
var_mean
|
|
std_mean
|
|
argmin
|
|
cummax
|
|
cummin
|
|
cumprod
|
|
cumsum
|
|
logcumsumexp
|
|
logsumexp
|
|
max
|
|
mean
|
|
median
|
|
min
|
|
mode
|
|
nanmean
|
|
nanmedian
|
|
nanquantile
|
|
nansum
|
|
quantile
|
|
std
|
|
std_mean
|
|
sum
|
|
var
|
|
var_mean
|
|
cumulative_trapezoid
|
|
diff
|
|
gradient
|
|
grid_sampler
|
|
grid_sampler_2d
|
|
grid_sampler_3d
|
|
lerp
|
|
trapezoid
|
|
trapz
|
|
arange
|
|
as_tensor
|
|
asarray
|
|
bartlett_window
|
|
blackman_window
|
|
empty
|
|
empty_like
|
|
empty_strided
|
|
eye
|
|
from_numpy
|
|
full
|
|
full_like
|
|
hamming_window
|
|
hann_window
|
|
kaiser_window
|
|
linspace
|
|
logspace
|
|
meshgrid
|
|
new
|
|
new_empty
|
|
new_empty
|
|
new_empty_strided
|
|
new_full
|
|
new_full
|
|
new_ones
|
|
new_ones
|
|
new_tensor
|
|
new_zeros
|
|
new_zeros
|
|
normal
|
|
ones
|
|
ones_like
|
|
rand
|
|
rand_like
|
|
randint
|
|
randint_like
|
|
randn
|
|
randn_like
|
|
random_
|
|
randperm
|
|
range
|
|
tensor
|
|
vander
|
|
zeros
|
|
zeros_like
|
|
adaptive_max_pool1d_with_indices
|
|
adaptive_max_pool2d_with_indices
|
|
adaptive_max_pool3d_with_indices
|
|
affine_grid
|
|
alpha_dropout
|
|
batch_norm_gather_stats_with_counts
|
|
batch_norm_stats
|
|
binary_cross_entropy
|
|
binary_cross_entropy_with_logits
|
|
cdist
|
|
channel_shuffle
|
|
constant_pad_nd
|
|
conv3d
|
|
convolution
|
|
dist
|
|
dropout3d
|
|
feature_alpha_dropout
|
|
feature_dropout
|
|
fractional_max_pool2d_with_indices
|
|
fractional_max_pool3d_with_indices
|
|
gru_cell
|
|
gumbel_softmax
|
|
kl_div
|
|
l1_loss
|
|
log_softmax
|
|
lp_pool1d
|
|
lp_pool2d
|
|
lp_pool3d
|
|
lstm_cell
|
|
margin_ranking_loss
|
|
max_pool1d_with_indices
|
|
max_pool2d_with_indices
|
|
max_pool3d_with_indices
|
|
max_unpool1d
|
|
max_unpool2d
|
|
max_unpool3d
|
|
multi_head_attention_forward
|
|
multi_margin_loss
|
|
multilabel_margin_loss
|
|
multilabel_soft_margin_loss
|
|
native_batch_norm
|
|
native_dropout
|
|
native_layer_norm
|
|
native_norm
|
|
nn.functional.adaptive_avg_pool1d
|
|
nn.functional.adaptive_avg_pool2d
|
|
nn.functional.adaptive_avg_pool3d
|
|
nn.functional.adaptive_max_pool1d
|
|
nn.functional.adaptive_max_pool2d
|
|
nn.functional.adaptive_max_pool3d
|
|
nn.functional.avg_pool1d
|
|
nn.functional.avg_pool2d
|
|
nn.functional.avg_pool3d
|
|
nn.functional.batch_norm
|
|
nn.functional.bilinear
|
|
nn.functional.conv_transpose1d
|
|
nn.functional.conv_transpose2d
|
|
nn.functional.conv_transpose3d
|
|
nn.functional.conv1d
|
|
nn.functional.conv2d
|
|
nn.functional.cosine_embedding_loss
|
|
nn.functional.cosine_similarity
|
|
nn.functional.cross_entropy
|
|
nn.functional.ctc_loss
|
|
nn.functional.dropout
|
|
nn.functional.dropout2d
|
|
nn.functional.embedding
|
|
nn.functional.embedding_bag
|
|
nn.functional.feature_alpha_dropout
|
|
nn.functional.fractional_max_pool2d
|
|
nn.functional.fractional_max_pool3d
|
|
nn.functional.gaussian_nll_loss
|
|
nn.functional.gelu
|
|
nn.functional.glu
|
|
nn.functional.grid_sample
|
|
nn.functional.group_norm
|
|
nn.functional.hardshrink
|
|
nn.functional.hardsigmoid
|
|
nn.functional.hardswish
|
|
nn.functional.hardtanh
|
|
nn.functional.hinge_embedding_loss
|
|
nn.functional.huber_loss
|
|
nn.functional.instance_norm
|
|
nn.functional.interpolate
|
|
nn.functional.kl_div
|
|
nn.functional.layer_norm
|
|
nn.functional.leaky_relu
|
|
nn.functional.linear
|
|
nn.functional.local_response_norm
|
|
nn.functional.logsigmoid
|
|
nn.functional.max_pool1d
|
|
nn.functional.max_pool2d
|
|
nn.functional.max_pool3d
|
|
nn.functional.mse_loss
|
|
nn.functional.nll_loss
|
|
nn.functional.normalize
|
|
nn.functional.one_hot
|
|
nn.functional.pad
|
|
nn.functional.pairwise_distance
|
|
nn.functional.pixel_shuffle
|
|
nn.functional.pixel_unshuffle
|
|
nn.functional.poisson_nll_loss
|
|
nn.functional.prelu
|
|
nn.functional.relu
|
|
nn.functional.relu6
|
|
nn.functional.softmin
|
|
nn.functional.softplus
|
|
nn.functional.softshrink
|
|
nn.functional.softsign
|
|
nn.functional.tanhshrink
|
|
nn.functional.threshold
|
|
nn.functional.unfold
|
|
nn.functional.upsample_bilinear
|
|
nn.functional.upsample_nearest
|
|
pdist
|
|
rnn_relu
|
|
rnn_relu_cell
|
|
rnn_tanh
|
|
rnn_tanh_cell
|
|
smooth_l1_loss
|
|
soft_margin_loss
|
|
softmax
|
|
triplet_margin_loss
|
|
triplet_margin_with_distance_loss
|
|
argsort
|
|
argwhere
|
|
bincount
|
|
bucketize
|
|
corrcoef
|
|
count_nonzero
|
|
cov
|
|
histc
|
|
histogram
|
|
histogramdd
|
|
isin
|
|
kthvalue
|
|
msort
|
|
nonzero
|
|
searchsorted
|
|
sort
|
|
topk
|
|
unique
|
|
unique_consecutive
|
|
clamp
|
|
clamp_max
|
|
clamp_min
|
|
where
|
|
diagonal_scatter
|
|
gather
|
|
scatter
|
|
scatter_add
|
|
scatter_reduce
|
|
select_scatter
|
|
slice_scatter
|
|
index_add
|
|
index_copy
|
|
index_fill
|
|
index_put
|
|
index_select
|
|
item
|
|
put
|
|
select
|
|
take
|
|
take_along_dim
|
|
bfloat16
|
|
bool
|
|
byte
|
|
char
|
|
clone
|
|
cpu
|
|
cuda
|
|
double
|
|
fill_
|
|
float
|
|
half
|
|
int
|
|
long
|
|
short
|
|
type_as
|
|
zero_
|
|
coalesce
|
|
dense_dim
|
|
sparse_coo_tensor
|
|
sparse_csr_tensor
|
|
to_dense
|
|
to_sparse
|
|
to_sparse_coo
|
|
complex
|
|
conj
|
|
conj_physical
|
|
real
|
|
resolve_conj
|
|
resolve_neg
|
|
view_as_complex
|
|
view_as_real
|
|
bernoulli
|
|
binomial
|
|
cauchy
|
|
exponential_
|
|
geometric_
|
|
multinomial
|
|
poisson
|
|
__getitem__
|
|
combinations
|
|
is_complex
|
|
is_floating_point
|
|
is_signed
|
|
sum_to_size
|
|
masked.amax
|
|
masked.amin
|
|
masked.log_softmax
|
|
masked.mean
|
|
masked.norm
|
|
masked.normalize
|
|
masked.prod
|
|
masked.softmax
|
|
masked.softmin
|
|
masked.std
|
|
masked.sum
|
|
masked.var
|
|
masked_fill
|
|
masked_scatter
|
|
masked_select
|
|
addcdiv
|
|
addcmul
|
|
allclose
|
|
equal
|