static Tensor |
torch._sparse_compressed_tensor_with_dims(long nnz,
long dense_dim,
long[] size,
long[] blocksize,
torch.ScalarType index_dtype,
TensorOptions options) |
static Tensor |
torch._sparse_compressed_tensor_with_dims(long nnz,
long dense_dim,
LongArrayRef size,
LongArrayRef blocksize,
torch.ScalarType index_dtype,
TensorOptions options) |
static Tensor |
torch.addmm_out(Tensor out,
Tensor self,
Tensor mat1,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.addmm_out(Tensor out,
Tensor self,
Tensor mat1,
Tensor mat2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha) |
static Tensor |
torch.addmm_outf(Tensor self,
Tensor mat1,
Tensor mat2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha,
Tensor out) |
static Tensor |
torch.addmm(Tensor self,
Tensor mat1,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.addmm(Tensor self,
Tensor mat1,
Tensor mat2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha) |
static Tensor |
torch.baddbmm_out(Tensor out,
Tensor self,
Tensor batch1,
Tensor batch2,
torch.ScalarType out_dtype) |
static Tensor |
torch.baddbmm_out(Tensor out,
Tensor self,
Tensor batch1,
Tensor batch2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha) |
static Tensor |
torch.baddbmm_outf(Tensor self,
Tensor batch1,
Tensor batch2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha,
Tensor out) |
static Tensor |
torch.baddbmm(Tensor self,
Tensor batch1,
Tensor batch2,
torch.ScalarType out_dtype) |
static Tensor |
torch.baddbmm(Tensor self,
Tensor batch1,
Tensor batch2,
torch.ScalarType out_dtype,
Scalar beta,
Scalar alpha) |
static Tensor |
torch.bmm_out(Tensor out,
Tensor self,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.bmm_outf(Tensor self,
Tensor mat2,
torch.ScalarType out_dtype,
Tensor out) |
static Tensor |
torch.bmm(Tensor self,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.cached_cast(torch.ScalarType to_type,
Tensor arg) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorArrayRef arg) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorArrayRef arg,
byte device_type) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorArrayRef arg,
torch.DeviceType device_type) |
static Tensor |
torch.cached_cast(torch.ScalarType to_type,
Tensor arg,
byte device_type) |
static TensorOptional |
torch.cached_cast(torch.ScalarType to_type,
TensorOptional arg) |
static TensorOptional |
torch.cached_cast(torch.ScalarType to_type,
TensorOptional arg,
byte device_type) |
static TensorOptional |
torch.cached_cast(torch.ScalarType to_type,
TensorOptional arg,
torch.DeviceType device_type) |
static Tensor |
torch.cached_cast(torch.ScalarType to_type,
Tensor arg,
torch.DeviceType device_type) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorVector arg) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorVector arg,
byte device_type) |
static TensorVector |
torch.cached_cast(torch.ScalarType to_type,
TensorVector arg,
torch.DeviceType device_type) |
static boolean |
torch.can_cast(torch.ScalarType from_,
torch.ScalarType to) |
static boolean |
torch.canCast(torch.ScalarType from,
torch.ScalarType to) |
static void |
torch.cast_and_store_from_BFloat16(torch.ScalarType dest_type,
Pointer ptr,
BFloat16 value) |
static void |
torch.cast_and_store_from_boolean(torch.ScalarType dest_type,
Pointer ptr,
boolean value) |
static void |
torch.cast_and_store_from_byte(torch.ScalarType dest_type,
Pointer ptr,
byte value) |
static void |
torch.cast_and_store_from_ComplexDouble(torch.ScalarType dest_type,
Pointer ptr,
DoubleComplex value) |
static void |
torch.cast_and_store_from_ComplexFloat(torch.ScalarType dest_type,
Pointer ptr,
FloatComplex value) |
static void |
torch.cast_and_store_from_double(torch.ScalarType dest_type,
Pointer ptr,
double value) |
static void |
torch.cast_and_store_from_float(torch.ScalarType dest_type,
Pointer ptr,
float value) |
static void |
torch.cast_and_store_from_Float8_e4m3fn(torch.ScalarType dest_type,
Pointer ptr,
Float8_e4m3fn value) |
static void |
torch.cast_and_store_from_Float8_e4m3fnuz(torch.ScalarType dest_type,
Pointer ptr,
Float8_e4m3fnuz value) |
static void |
torch.cast_and_store_from_Float8_e5m2(torch.ScalarType dest_type,
Pointer ptr,
Float8_e5m2 value) |
static void |
torch.cast_and_store_from_Float8_e5m2fnuz(torch.ScalarType dest_type,
Pointer ptr,
Float8_e5m2fnuz value) |
static void |
torch.cast_and_store_from_Half(torch.ScalarType dest_type,
Pointer ptr,
Half value) |
static void |
torch.cast_and_store_from_int(torch.ScalarType dest_type,
Pointer ptr,
int value) |
static void |
torch.cast_and_store_from_long(torch.ScalarType dest_type,
Pointer ptr,
long value) |
static void |
torch.cast_and_store_from_qint8(torch.ScalarType dest_type,
Pointer ptr,
qint8 value) |
static void |
torch.cast_and_store_from_quint2x4(torch.ScalarType dest_type,
Pointer ptr,
quint2x4 value) |
static void |
torch.cast_and_store_from_quint32(torch.ScalarType dest_type,
Pointer ptr,
qint32 value) |
static void |
torch.cast_and_store_from_quint4x2(torch.ScalarType dest_type,
Pointer ptr,
quint4x2 value) |
static void |
torch.cast_and_store_from_quint8(torch.ScalarType dest_type,
Pointer ptr,
quint8 value) |
static void |
torch.cast_and_store_from_short(torch.ScalarType dest_type,
Pointer ptr,
short value) |
static TensorImplVector |
torch.checked_dense_tensor_list_unwrap(TensorArrayRef tensors,
BytePointer name,
int pos,
torch.DeviceType device_type,
torch.ScalarType scalar_type) |
static TensorImplVector |
torch.checked_dense_tensor_list_unwrap(TensorArrayRef tensors,
String name,
int pos,
byte device_type,
torch.ScalarType scalar_type) |
static TensorImplVector |
torch.checked_dense_tensor_list_unwrap(TensorVector tensors,
BytePointer name,
int pos,
torch.DeviceType device_type,
torch.ScalarType scalar_type) |
static TensorImplVector |
torch.checked_dense_tensor_list_unwrap(TensorVector tensors,
String name,
int pos,
byte device_type,
torch.ScalarType scalar_type) |
static void |
torch.checkScalarType(BytePointer c,
TensorArg t,
torch.ScalarType s) |
static void |
torch.checkScalarType(String c,
TensorArg t,
torch.ScalarType s) |
static torch.ScalarType |
torch.compute_desired_dtype(torch.ScalarType scalar_type) |
static TensorOptions |
torch.dtype(torch.ScalarType dtype) |
static long |
torch.elementSize(torch.ScalarType t) |
static TensorBase |
torch.empty_cpu(long[] size,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_cpu(long[] size,
torch.ScalarType dtype,
boolean pin_memory,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_cpu(LongArrayRef size,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_cpu(LongArrayRef size,
torch.ScalarType dtype,
boolean pin_memory,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_generic_symint(SymIntArrayRef size,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_generic(long[] size,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_generic(LongArrayRef size,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_meta(long[] size,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_meta(long[] size,
torch.ScalarType dtype,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_meta(LongArrayRef size,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_meta(LongArrayRef size,
torch.ScalarType dtype,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_strided_cpu(long[] size,
long[] stride,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_strided_cpu(long[] size,
long[] stride,
torch.ScalarType dtype,
boolean pin_memory) |
static TensorBase |
torch.empty_strided_cpu(LongArrayRef size,
LongArrayRef stride,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_strided_cpu(LongArrayRef size,
LongArrayRef stride,
torch.ScalarType dtype,
boolean pin_memory) |
static TensorBase |
torch.empty_strided_generic(long[] size,
long[] stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static TensorBase |
torch.empty_strided_generic(LongArrayRef size,
LongArrayRef stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static TensorBase |
torch.empty_strided_meta(long[] size,
long[] stride,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_strided_meta(LongArrayRef size,
LongArrayRef stride,
torch.ScalarType dtype) |
static TensorBase |
torch.empty_strided_symint_generic(SymIntArrayRef size,
SymIntArrayRef stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static TensorBase |
torch.empty_strided_symint_meta(SymIntArrayRef size,
SymIntArrayRef stride,
torch.ScalarType dtype) |
static boolean |
torch.equals(torch.ScalarType t,
TypeMeta m)
convenience: equality across TypeMeta/ScalarType conversion
|
static boolean |
torch.equals(TypeMeta m,
torch.ScalarType t) |
static BFloat16 |
torch.fetch_and_cast_to_BFloat16(torch.ScalarType src_type,
Pointer ptr) |
static boolean |
torch.fetch_and_cast_to_boolean(torch.ScalarType src_type,
Pointer ptr) |
static byte |
torch.fetch_and_cast_to_byte(torch.ScalarType src_type,
Pointer ptr) |
static DoubleComplex |
torch.fetch_and_cast_to_ComplexDouble(torch.ScalarType src_type,
Pointer ptr) |
static FloatComplex |
torch.fetch_and_cast_to_ComplexFloat(torch.ScalarType src_type,
Pointer ptr) |
static double |
torch.fetch_and_cast_to_double(torch.ScalarType src_type,
Pointer ptr) |
static float |
torch.fetch_and_cast_to_float(torch.ScalarType src_type,
Pointer ptr) |
static Float8_e4m3fn |
torch.fetch_and_cast_to_Float8_e4m3fn(torch.ScalarType src_type,
Pointer ptr) |
static Float8_e4m3fnuz |
torch.fetch_and_cast_to_Float8_e4m3fnuz(torch.ScalarType src_type,
Pointer ptr) |
static Float8_e5m2 |
torch.fetch_and_cast_to_Float8_e5m2(torch.ScalarType src_type,
Pointer ptr) |
static Float8_e5m2fnuz |
torch.fetch_and_cast_to_Float8_e5m2fnuz(torch.ScalarType src_type,
Pointer ptr) |
static Half |
torch.fetch_and_cast_to_Half(torch.ScalarType src_type,
Pointer ptr) |
static int |
torch.fetch_and_cast_to_int(torch.ScalarType src_type,
Pointer ptr) |
static long |
torch.fetch_and_cast_to_long(torch.ScalarType src_type,
Pointer ptr) |
static qint8 |
torch.fetch_and_cast_to_qint8(torch.ScalarType src_type,
Pointer ptr) |
static quint2x4 |
torch.fetch_and_cast_to_quint2x4(torch.ScalarType src_type,
Pointer ptr) |
static qint32 |
torch.fetch_and_cast_to_quint32(torch.ScalarType src_type,
Pointer ptr) |
static quint4x2 |
torch.fetch_and_cast_to_quint4x2(torch.ScalarType src_type,
Pointer ptr) |
static quint8 |
torch.fetch_and_cast_to_quint8(torch.ScalarType src_type,
Pointer ptr) |
static short |
torch.fetch_and_cast_to_short(torch.ScalarType src_type,
Pointer ptr) |
static int |
torch_cuda.getCudnnDataTypeFromScalarType(torch.ScalarType dtype)
A variant of OptionalStreamGuard that is specialized for CUDA.
|
static StringPair |
torch.getDtypeNames(torch.ScalarType scalarType) |
static int |
torch_nccl.getNcclDataType(torch.ScalarType type) |
static boolean |
torch.isBarebonesUnsignedType(torch.ScalarType t) |
static boolean |
torch.isBitsType(torch.ScalarType t) |
static boolean |
torch.isComplexType(torch.ScalarType t) |
static boolean |
torch.isFloat8Type(torch.ScalarType t) |
static boolean |
torch.isFloatingType(torch.ScalarType t) |
static boolean |
torch.isIntegralType(torch.ScalarType t,
boolean includeBool) |
static boolean |
torch.isQIntType(torch.ScalarType t) |
static boolean |
torch.isReducedFloatingType(torch.ScalarType t) |
static boolean |
torch.isSignedType(torch.ScalarType t) |
static boolean |
torch.isUnderlying(torch.ScalarType type,
torch.ScalarType qtype) |
static Tensor |
torch.mm_out(Tensor out,
Tensor self,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.mm_outf(Tensor self,
Tensor mat2,
torch.ScalarType out_dtype,
Tensor out) |
static Tensor |
torch.mm(Tensor self,
Tensor mat2,
torch.ScalarType out_dtype) |
static Tensor |
torch.norm_out(Tensor out,
Tensor self,
ScalarOptional p,
DimnameArrayRef dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm_out(Tensor out,
Tensor self,
ScalarOptional p,
DimnameVector dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm_out(Tensor out,
Tensor self,
ScalarOptional p,
long[] dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm_out(Tensor out,
Tensor self,
ScalarOptional p,
LongArrayRef dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm_out(Tensor out,
Tensor self,
ScalarOptional p,
torch.ScalarType dtype) |
static Tensor |
torch.norm_outf(Tensor self,
ScalarOptional p,
DimnameArrayRef dim,
boolean keepdim,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.norm_outf(Tensor self,
ScalarOptional p,
DimnameVector dim,
boolean keepdim,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.norm_outf(Tensor self,
ScalarOptional p,
long[] dim,
boolean keepdim,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.norm_outf(Tensor self,
ScalarOptional p,
LongArrayRef dim,
boolean keepdim,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.norm_outf(Tensor self,
ScalarOptional p,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.norm(Tensor self,
ScalarOptional p,
DimnameArrayRef dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm(Tensor self,
ScalarOptional p,
DimnameVector dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm(Tensor self,
ScalarOptional p,
long[] dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm(Tensor self,
ScalarOptional p,
LongArrayRef dim,
boolean keepdim,
torch.ScalarType dtype) |
static Tensor |
torch.norm(Tensor self,
ScalarOptional p,
torch.ScalarType dtype) |
static boolean |
torch.notEquals(torch.ScalarType t,
TypeMeta m) |
static boolean |
torch.notEquals(TypeMeta m,
torch.ScalarType t) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
Tensor nextArg) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorArrayRef list) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorArrayRef list,
byte device_type) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorArrayRef list,
torch.DeviceType device_type) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
Tensor nextArg,
byte device_type) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
Tensor nextArg,
torch.DeviceType device_type)
Logic to extract the promote type from any Tensor or TensorList args.
|
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorVector list) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorVector list,
byte device_type) |
static torch.ScalarType |
torch.prioritize(torch.ScalarType current,
TensorVector list,
torch.DeviceType device_type) |
static torch.ScalarType |
torch.promote_type(torch.ScalarType current,
byte device_type) |
static torch.ScalarType |
torch.promote_type(torch.ScalarType current,
torch.DeviceType device_type) |
static torch.ScalarType |
torch.promote_types(torch.ScalarType type1,
torch.ScalarType type2) |
static torch.ScalarType |
torch.promoteTypes(torch.ScalarType a,
torch.ScalarType b) |
static Tensor |
torch.quantize_per_channel_out(Tensor out,
Tensor self,
Tensor scales,
Tensor zero_points,
long axis,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_channel_outf(Tensor self,
Tensor scales,
Tensor zero_points,
long axis,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.quantize_per_channel(Tensor self,
Tensor scales,
Tensor zero_points,
long axis,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_tensor_dynamic_out(Tensor out,
Tensor self,
torch.ScalarType dtype,
boolean reduce_range) |
static Tensor |
torch.quantize_per_tensor_dynamic_outf(Tensor self,
torch.ScalarType dtype,
boolean reduce_range,
Tensor out) |
static Tensor |
torch.quantize_per_tensor_dynamic(Tensor self,
torch.ScalarType dtype,
boolean reduce_range) |
static void |
torch.quantize_per_tensor_out(TensorArrayRef out,
TensorArrayRef tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_tensor_out(Tensor out,
Tensor self,
double scale,
long zero_point,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_tensor_out(Tensor out,
Tensor self,
Tensor scale,
Tensor zero_point,
torch.ScalarType dtype) |
static void |
torch.quantize_per_tensor_out(TensorVector out,
TensorVector tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype) |
static void |
torch.quantize_per_tensor_outf(TensorArrayRef tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype,
TensorArrayRef out) |
static Tensor |
torch.quantize_per_tensor_outf(Tensor self,
double scale,
long zero_point,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.quantize_per_tensor_outf(Tensor self,
Tensor scale,
Tensor zero_point,
torch.ScalarType dtype,
Tensor out) |
static void |
torch.quantize_per_tensor_outf(TensorVector tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype,
TensorVector out) |
static TensorVector |
torch.quantize_per_tensor(TensorArrayRef tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_tensor(Tensor self,
double scale,
long zero_point,
torch.ScalarType dtype) |
static Tensor |
torch.quantize_per_tensor(Tensor self,
Tensor scale,
Tensor zero_point,
torch.ScalarType dtype) |
static TensorVector |
torch.quantize_per_tensor(TensorVector tensors,
Tensor scales,
Tensor zero_points,
torch.ScalarType dtype) |
static TypeMeta |
torch.scalarTypeToTypeMeta(torch.ScalarType scalar_type)
convert ScalarType enum values to TypeMeta handles
|
static void |
torch.set_autocast_cpu_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_dtype(torch.DeviceType device_type,
torch.ScalarType dtype) |
static void |
torch.set_autocast_gpu_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_hpu_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_ipu_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_mtia_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_privateuseone_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_xla_dtype(torch.ScalarType dtype)
Deprecated.
|
static void |
torch.set_autocast_xpu_dtype(torch.ScalarType dtype)
Deprecated.
|
static ScalarTypeOptional |
torch.set_opt_dtype(torch.ScalarType to_type,
ScalarTypeOptional dtype)
Logic to flip an output dtype flag.
|
static Pointer |
torch.shiftLeft(Pointer stream,
torch.ScalarType scalar_type) |
static boolean |
torch.should_include_kernel_dtype(BytePointer arg0,
torch.ScalarType arg1)
The method should_include_kernel_dtype() returns true/false
based on whether the switching code for a specific dtype should be
included based on build time constants generated from tracing model
execution.
|
static boolean |
torch.should_include_kernel_dtype(String arg0,
torch.ScalarType arg1) |
static torch.ScalarType |
torch.toComplexType(torch.ScalarType t) |
static torch.ScalarType |
torch.toQIntType(torch.ScalarType t) |
static torch.ScalarType |
torch.toRealValueType(torch.ScalarType t) |
static BytePointer |
torch.toString(torch.ScalarType t) |
static torch.ScalarType |
torch.toUnderlying(torch.ScalarType t) |
static Tensor |
torch.view_copy_out(Tensor out,
Tensor self,
torch.ScalarType dtype) |
static Tensor |
torch.view_copy_outf(Tensor self,
torch.ScalarType dtype,
Tensor out) |
static Tensor |
torch.view_copy(Tensor self,
torch.ScalarType dtype) |