torchwrench package¶
Collection of functions and modules to help development in PyTorch.
Subpackages¶
- torchwrench.core package
- Submodules
- torchwrench.core.config module
- torchwrench.core.dtype module
- torchwrench.core.dtype_enum module
- torchwrench.core.dtype_enum.DTypeEnum
- torchwrench.core.dtype_enum.enum_dtype_to_str
- torchwrench.core.dtype_enum.enum_dtype_to_torch_dtype
- torchwrench.core.dtype_enum.str_to_enum_dtype
- torchwrench.core.dtype_enum.str_to_torch_dtype
- torchwrench.core.dtype_enum.torch_dtype_to_enum_dtype
- torchwrench.core.dtype_enum.torch_dtype_to_str
- torchwrench.core.make module
- torchwrench.core.make.as_device
- torchwrench.core.make.as_dtype
- torchwrench.core.make.as_generator
- torchwrench.core.make.get_default_device
- torchwrench.core.make.get_default_dtype
- torchwrench.core.make.get_default_generator
- torchwrench.core.make.set_default_dtype
- torchwrench.core.make.set_default_generator
- torchwrench.core.packaging module
- torchwrench.core.version module
- Submodules
- torchwrench.extras package
- Subpackages
- torchwrench.extras.hdf package
- torchwrench.extras.numpy package
- torchwrench.extras.numpy.InvalidTorchDType
- torchwrench.extras.numpy.ShapeDTypeInfo
- torchwrench.extras.numpy.get_default_numpy_dtype
- torchwrench.extras.numpy.is_numpy_bool_array
- torchwrench.extras.numpy.is_numpy_integral_array
- torchwrench.extras.numpy.is_numpy_number_like
- torchwrench.extras.numpy.is_numpy_scalar_like
- torchwrench.extras.numpy.is_numpy_str_array
- torchwrench.extras.numpy.logical_and_lst
- torchwrench.extras.numpy.logical_or_lst
- torchwrench.extras.numpy.merge_numpy_dtypes
- torchwrench.extras.numpy.merge_torch_dtypes
- torchwrench.extras.numpy.ndarray_to_tensor
- torchwrench.extras.numpy.numpy_all_eq
- torchwrench.extras.numpy.numpy_all_ne
- torchwrench.extras.numpy.numpy_complex_dtype_to_float_dtype
- torchwrench.extras.numpy.numpy_dtype_to_fill_value
- torchwrench.extras.numpy.numpy_dtype_to_torch_dtype
- torchwrench.extras.numpy.numpy_is_complex
- torchwrench.extras.numpy.numpy_is_complex_dtype
- torchwrench.extras.numpy.numpy_is_floating_point
- torchwrench.extras.numpy.numpy_item
- torchwrench.extras.numpy.numpy_to_tensor
- torchwrench.extras.numpy.numpy_topk
- torchwrench.extras.numpy.numpy_view_as_complex
- torchwrench.extras.numpy.numpy_view_as_real
- torchwrench.extras.numpy.scan_numpy_dtype
- torchwrench.extras.numpy.scan_shape_dtypes
- torchwrench.extras.numpy.scan_torch_dtype
- torchwrench.extras.numpy.tensor_to_ndarray
- torchwrench.extras.numpy.tensor_to_numpy
- torchwrench.extras.numpy.to_ndarray
- torchwrench.extras.numpy.to_numpy
- torchwrench.extras.numpy.torch_dtype_to_numpy_dtype
- Submodules
- torchwrench.extras.pandas package
- torchwrench.extras.speechbrain package
- torchwrench.extras.yaml package
- Submodules
- torchwrench.extras.colorlog module
- torchwrench.extras.safetensors module
- torchwrench.extras.tensorboard module
- torchwrench.extras.torchaudio module
- torchwrench.extras.torchaudio.AudioMetaDataDict
- torchwrench.extras.torchaudio.CodecConfig
- torchwrench.extras.torchaudio.audio_metadata_to_dict
- torchwrench.extras.torchaudio.dump_audio
- torchwrench.extras.torchaudio.dump_with_torchaudio
- torchwrench.extras.torchaudio.load_audio
- torchwrench.extras.torchaudio.load_with_torchaudio
- Subpackages
- torchwrench.hub package
- torchwrench.nn package
- torchwrench.nn.Abs
- torchwrench.nn.AdaptiveAvgPool1d
- torchwrench.nn.AdaptiveAvgPool2d
- torchwrench.nn.AdaptiveAvgPool3d
- torchwrench.nn.AdaptiveLogSoftmaxWithLoss
- torchwrench.nn.AdaptiveMaxPool1d
- torchwrench.nn.AdaptiveMaxPool2d
- torchwrench.nn.AdaptiveMaxPool3d
- torchwrench.nn.AlphaDropout
- torchwrench.nn.Angle
- torchwrench.nn.AsTensor
- torchwrench.nn.AvgPool1d
- torchwrench.nn.AvgPool2d
- torchwrench.nn.AvgPool3d
- torchwrench.nn.BCELoss
- torchwrench.nn.BCEWithLogitsLoss
- torchwrench.nn.BatchNorm1d
- torchwrench.nn.BatchNorm2d
- torchwrench.nn.BatchNorm3d
- torchwrench.nn.Bilinear
- torchwrench.nn.CELU
- torchwrench.nn.CTCLoss
- torchwrench.nn.ChannelShuffle
- torchwrench.nn.ConstantPad1d
- torchwrench.nn.ConstantPad2d
- torchwrench.nn.ConstantPad3d
- torchwrench.nn.Conv1d
- torchwrench.nn.Conv2d
- torchwrench.nn.Conv3d
- torchwrench.nn.ConvTranspose1d
- torchwrench.nn.ConvTranspose2d
- torchwrench.nn.ConvTranspose3d
- torchwrench.nn.CosineEmbeddingLoss
- torchwrench.nn.CosineSimilarity
- torchwrench.nn.CropDim
- torchwrench.nn.CropDims
- torchwrench.nn.CrossEntropyLoss
- torchwrench.nn.CrossMapLRN2d
- torchwrench.nn.Dropout
- torchwrench.nn.Dropout2d
- torchwrench.nn.Dropout3d
- torchwrench.nn.ELU
- torchwrench.nn.EModule
- torchwrench.nn.EModuleDict
- torchwrench.nn.EModuleList
- torchwrench.nn.EModulePartial
- torchwrench.nn.ESequential
- torchwrench.nn.Embedding
- torchwrench.nn.EmbeddingBag
- torchwrench.nn.Exp
- torchwrench.nn.Exp2
- torchwrench.nn.FFT
- torchwrench.nn.FeatureAlphaDropout
- torchwrench.nn.Flatten
- torchwrench.nn.Fold
- torchwrench.nn.FractionalMaxPool2d
- torchwrench.nn.FractionalMaxPool3d
- torchwrench.nn.GELU
- torchwrench.nn.GLU
- torchwrench.nn.GRU
- torchwrench.nn.GRUCell
- torchwrench.nn.GaussianNLLLoss
- torchwrench.nn.GroupNorm
- torchwrench.nn.Hardshrink
- torchwrench.nn.Hardsigmoid
- torchwrench.nn.Hardswish
- torchwrench.nn.Hardtanh
- torchwrench.nn.HingeEmbeddingLoss
- torchwrench.nn.HuberLoss
- torchwrench.nn.IFFT
- torchwrench.nn.Identity
- torchwrench.nn.Imag
- torchwrench.nn.IndexToName
- torchwrench.nn.IndexToOnehot
- torchwrench.nn.IndicesToMultihot
- torchwrench.nn.IndicesToMultinames
- torchwrench.nn.InstanceNorm1d
- torchwrench.nn.InstanceNorm2d
- torchwrench.nn.InstanceNorm3d
- torchwrench.nn.KLDivLoss
- torchwrench.nn.L1Loss
- torchwrench.nn.LPPool1d
- torchwrench.nn.LPPool2d
- torchwrench.nn.LSTM
- torchwrench.nn.LSTMCell
- torchwrench.nn.LayerNorm
- torchwrench.nn.LazyBatchNorm1d
- torchwrench.nn.LazyBatchNorm2d
- torchwrench.nn.LazyBatchNorm3d
- torchwrench.nn.LazyConv1d
- torchwrench.nn.LazyConv2d
- torchwrench.nn.LazyConv3d
- torchwrench.nn.LazyConvTranspose1d
- torchwrench.nn.LazyConvTranspose2d
- torchwrench.nn.LazyConvTranspose3d
- torchwrench.nn.LazyInstanceNorm1d
- torchwrench.nn.LazyInstanceNorm2d
- torchwrench.nn.LazyInstanceNorm3d
- torchwrench.nn.LazyLinear
- torchwrench.nn.LeakyReLU
- torchwrench.nn.Linear
- torchwrench.nn.LocalResponseNorm
- torchwrench.nn.Log
- torchwrench.nn.Log10
- torchwrench.nn.Log2
- torchwrench.nn.LogSigmoid
- torchwrench.nn.LogSoftmax
- torchwrench.nn.LogSoftmaxMultidim
- torchwrench.nn.MSELoss
- torchwrench.nn.MarginRankingLoss
- torchwrench.nn.MaskedMean
- torchwrench.nn.MaskedSum
- torchwrench.nn.Max
- torchwrench.nn.MaxPool1d
- torchwrench.nn.MaxPool2d
- torchwrench.nn.MaxPool3d
- torchwrench.nn.MaxUnpool1d
- torchwrench.nn.MaxUnpool2d
- torchwrench.nn.MaxUnpool3d
- torchwrench.nn.Mean
- torchwrench.nn.Min
- torchwrench.nn.Mish
- torchwrench.nn.Module
- Variables
- T_destination
- add_module
- apply
- bfloat16
- buffers
- call_super_init
- children
- compile
- cpu
- cuda
- double
- dump_patches
- eval
- extra_repr
- float
- forward
- get_buffer
- get_extra_state
- get_parameter
- get_submodule
- half
- ipu
- load_state_dict
- modules
- mtia
- named_buffers
- named_children
- named_modules
- named_parameters
- parameters
- register_backward_hook
- register_buffer
- register_forward_hook
- register_forward_pre_hook
- register_full_backward_hook
- register_full_backward_pre_hook
- register_load_state_dict_post_hook
- register_load_state_dict_pre_hook
- register_module
- register_parameter
- register_state_dict_post_hook
- register_state_dict_pre_hook
- requires_grad_
- set_extra_state
- set_submodule
- share_memory
- state_dict
- to
- to_empty
- train
- training
- type
- xpu
- zero_grad
- torchwrench.nn.ModuleDict
- torchwrench.nn.ModuleList
- torchwrench.nn.ModulePartial
- torchwrench.nn.MoveToRec
- torchwrench.nn.MultiIndicesToMultihot
- torchwrench.nn.MultiIndicesToMultinames
- torchwrench.nn.MultiLabelMarginLoss
- torchwrench.nn.MultiLabelSoftMarginLoss
- torchwrench.nn.MultiMarginLoss
- torchwrench.nn.MultiheadAttention
- torchwrench.nn.MultihotToIndices
- torchwrench.nn.MultihotToMultiIndices
- torchwrench.nn.MultihotToMultinames
- torchwrench.nn.MultilabelToPowerset
- torchwrench.nn.MultinamesToIndices
- torchwrench.nn.MultinamesToMultiIndices
- torchwrench.nn.MultinamesToMultihot
- torchwrench.nn.NDArrayToTensor
- torchwrench.nn.NLLLoss
- torchwrench.nn.NameToIndex
- torchwrench.nn.NameToOnehot
- torchwrench.nn.Normalize
- torchwrench.nn.OnehotToIndex
- torchwrench.nn.OnehotToName
- torchwrench.nn.PReLU
- torchwrench.nn.PadAndCropDim
- torchwrench.nn.PadAndStackRec
- torchwrench.nn.PadDim
- torchwrench.nn.PadDims
- torchwrench.nn.PairwiseDistance
- torchwrench.nn.ParameterDict
- torchwrench.nn.ParameterList
- torchwrench.nn.Permute
- torchwrench.nn.PixelShuffle
- torchwrench.nn.PixelUnshuffle
- torchwrench.nn.PoissonNLLLoss
- torchwrench.nn.PositionalEncoding
- torchwrench.nn.Pow
- torchwrench.nn.PowersetToMultilabel
- torchwrench.nn.ProbsToIndex
- torchwrench.nn.ProbsToIndices
- torchwrench.nn.ProbsToMultiIndices
- torchwrench.nn.ProbsToMultihot
- torchwrench.nn.ProbsToMultinames
- torchwrench.nn.ProbsToName
- torchwrench.nn.ProbsToOnehot
- torchwrench.nn.RNN
- torchwrench.nn.RNNBase
- torchwrench.nn.RNNCell
- torchwrench.nn.RNNCellBase
- torchwrench.nn.RReLU
- torchwrench.nn.ReLU
- torchwrench.nn.ReLU6
- torchwrench.nn.Real
- torchwrench.nn.ReflectionPad1d
- torchwrench.nn.ReflectionPad2d
- torchwrench.nn.ReflectionPad3d
- torchwrench.nn.Repeat
- torchwrench.nn.RepeatInterleave
- torchwrench.nn.RepeatInterleaveNd
- torchwrench.nn.ReplicationPad1d
- torchwrench.nn.ReplicationPad2d
- torchwrench.nn.ReplicationPad3d
- torchwrench.nn.ResampleNearestFreqs
- torchwrench.nn.ResampleNearestRates
- torchwrench.nn.ResampleNearestSteps
- torchwrench.nn.Reshape
- torchwrench.nn.SELU
- torchwrench.nn.Sequential
- torchwrench.nn.Shuffled
- torchwrench.nn.SiLU
- torchwrench.nn.Sigmoid
- torchwrench.nn.SmoothL1Loss
- torchwrench.nn.SoftMarginLoss
- torchwrench.nn.Softmax
- torchwrench.nn.Softmax2d
- torchwrench.nn.SoftmaxMultidim
- torchwrench.nn.Softmin
- torchwrench.nn.Softplus
- torchwrench.nn.Softshrink
- torchwrench.nn.Softsign
- torchwrench.nn.Sort
- torchwrench.nn.Squeeze
- torchwrench.nn.SyncBatchNorm
- torchwrench.nn.TFlatten
- torchwrench.nn.Tanh
- torchwrench.nn.Tanhshrink
- torchwrench.nn.TensorTo
- torchwrench.nn.TensorToNDArray
- torchwrench.nn.Threshold
- torchwrench.nn.ToItem
- torchwrench.nn.ToList
- torchwrench.nn.ToNDArray
- torchwrench.nn.ToTensor
- torchwrench.nn.TopP
- torchwrench.nn.Topk
- torchwrench.nn.TransformDrop
- torchwrench.nn.Transformer
- torchwrench.nn.TransformerDecoder
- torchwrench.nn.TransformerDecoderLayer
- torchwrench.nn.TransformerEncoder
- torchwrench.nn.TransformerEncoderLayer
- torchwrench.nn.Transpose
- torchwrench.nn.TripletMarginLoss
- torchwrench.nn.TripletMarginWithDistanceLoss
- torchwrench.nn.Unflatten
- torchwrench.nn.Unfold
- torchwrench.nn.Unsqueeze
- torchwrench.nn.Upsample
- torchwrench.nn.UpsamplingBilinear2d
- torchwrench.nn.UpsamplingNearest2d
- torchwrench.nn.View
- torchwrench.nn.ViewAsComplex
- torchwrench.nn.ViewAsReal
- torchwrench.nn.ZeroPad2d
- Subpackages
- torchwrench.nn.functional package
- torchwrench.nn.functional.activity_to_segments
- torchwrench.nn.functional.activity_to_segments_list
- torchwrench.nn.functional.all_eq
- torchwrench.nn.functional.all_ne
- torchwrench.nn.functional.arange
- torchwrench.nn.functional.as_device
- torchwrench.nn.functional.as_dtype
- torchwrench.nn.functional.as_generator
- torchwrench.nn.functional.as_tensor
- torchwrench.nn.functional.average_power
- torchwrench.nn.functional.cat
- torchwrench.nn.functional.cat_padded_batch
- torchwrench.nn.functional.checksum_any
- torchwrench.nn.functional.concat
- torchwrench.nn.functional.count_parameters
- torchwrench.nn.functional.crop_dim
- torchwrench.nn.functional.crop_dims
- torchwrench.nn.functional.deep_equal
- torchwrench.nn.functional.empty
- torchwrench.nn.functional.equal
- torchwrench.nn.functional.find
- torchwrench.nn.functional.flatten
- torchwrench.nn.functional.full
- torchwrench.nn.functional.generate_square_subsequent_mask
- torchwrench.nn.functional.get_default_device
- torchwrench.nn.functional.get_default_dtype
- torchwrench.nn.functional.get_default_generator
- torchwrench.nn.functional.get_inverse_perm
- torchwrench.nn.functional.get_ndim
- torchwrench.nn.functional.get_perm_indices
- torchwrench.nn.functional.get_shape
- torchwrench.nn.functional.identity
- torchwrench.nn.functional.index_to_name
- torchwrench.nn.functional.index_to_onehot
- torchwrench.nn.functional.indices_to_multihot
- torchwrench.nn.functional.indices_to_multinames
- torchwrench.nn.functional.initial_seed
- torchwrench.nn.functional.insert_at_indices
- torchwrench.nn.functional.is_complex
- torchwrench.nn.functional.is_convertible_to_tensor
- torchwrench.nn.functional.is_floating_point
- torchwrench.nn.functional.is_full
- torchwrench.nn.functional.is_sorted
- torchwrench.nn.functional.is_stackable
- torchwrench.nn.functional.is_tensor
- torchwrench.nn.functional.is_unique
- torchwrench.nn.functional.lengths_to_non_pad_mask
- torchwrench.nn.functional.lengths_to_pad_mask
- torchwrench.nn.functional.lengths_to_ratios
- torchwrench.nn.functional.log_softmax_multidim
- torchwrench.nn.functional.manual_seed
- torchwrench.nn.functional.masked_equal
- torchwrench.nn.functional.masked_mean
- torchwrench.nn.functional.masked_sum
- torchwrench.nn.functional.matmul
- torchwrench.nn.functional.move_to
- torchwrench.nn.functional.move_to_rec
- torchwrench.nn.functional.mse
- torchwrench.nn.functional.multi_indices_to_multihot
- torchwrench.nn.functional.multi_indices_to_multinames
- torchwrench.nn.functional.multihot_to_indices
- torchwrench.nn.functional.multihot_to_multi_indices
- torchwrench.nn.functional.multihot_to_multinames
- torchwrench.nn.functional.multilabel_to_powerset
- torchwrench.nn.functional.multinames_to_indices
- torchwrench.nn.functional.multinames_to_multi_indices
- torchwrench.nn.functional.multinames_to_multihot
- torchwrench.nn.functional.name_to_index
- torchwrench.nn.functional.name_to_onehot
- torchwrench.nn.functional.ndim
- torchwrench.nn.functional.nelement
- torchwrench.nn.functional.no_grad
- torchwrench.nn.functional.non_pad_mask_to_lengths
- torchwrench.nn.functional.non_pad_mask_to_ratios
- torchwrench.nn.functional.one_hot
- torchwrench.nn.functional.onehot_to_index
- torchwrench.nn.functional.onehot_to_name
- torchwrench.nn.functional.ones
- torchwrench.nn.functional.pad_and_crop_dim
- torchwrench.nn.functional.pad_and_stack_rec
- torchwrench.nn.functional.pad_dim
- torchwrench.nn.functional.pad_dims
- torchwrench.nn.functional.pad_mask_to_lengths
- torchwrench.nn.functional.pad_mask_to_ratios
- torchwrench.nn.functional.powerset_to_multilabel
- torchwrench.nn.functional.probs_to_index
- torchwrench.nn.functional.probs_to_indices
- torchwrench.nn.functional.probs_to_multi_indices
- torchwrench.nn.functional.probs_to_multihot
- torchwrench.nn.functional.probs_to_multinames
- torchwrench.nn.functional.probs_to_name
- torchwrench.nn.functional.probs_to_onehot
- torchwrench.nn.functional.prod
- torchwrench.nn.functional.rand
- torchwrench.nn.functional.randint
- torchwrench.nn.functional.randn
- torchwrench.nn.functional.randperm
- torchwrench.nn.functional.randperm_diff
- torchwrench.nn.functional.ranks
- torchwrench.nn.functional.ratios_to_lengths
- torchwrench.nn.functional.ratios_to_non_pad_mask
- torchwrench.nn.functional.ratios_to_pad_mask
- torchwrench.nn.functional.recursive_to
- torchwrench.nn.functional.remove_at_indices
- torchwrench.nn.functional.repeat_interleave_nd
- torchwrench.nn.functional.resample_nearest_freqs
- torchwrench.nn.functional.resample_nearest_rates
- torchwrench.nn.functional.resample_nearest_steps
- torchwrench.nn.functional.rmse
- torchwrench.nn.functional.seed
- torchwrench.nn.functional.segments_list_to_activity
- torchwrench.nn.functional.segments_to_activity
- torchwrench.nn.functional.segments_to_segments_list
- torchwrench.nn.functional.set_default_dtype
- torchwrench.nn.functional.set_default_generator
- torchwrench.nn.functional.shape
- torchwrench.nn.functional.shuffled
- torchwrench.nn.functional.softmax_multidim
- torchwrench.nn.functional.split
- torchwrench.nn.functional.squeeze
- torchwrench.nn.functional.squeeze_
- torchwrench.nn.functional.squeeze_copy
- torchwrench.nn.functional.stack
- torchwrench.nn.functional.tensor_to_lengths
- torchwrench.nn.functional.tensor_to_non_pad_mask
- torchwrench.nn.functional.tensor_to_pad_mask
- torchwrench.nn.functional.tensor_to_tensors_list
- torchwrench.nn.functional.tensors_list_to_lengths
- torchwrench.nn.functional.to_item
- torchwrench.nn.functional.to_tensor
- torchwrench.nn.functional.top_k
- torchwrench.nn.functional.top_p
- torchwrench.nn.functional.topk
- torchwrench.nn.functional.transform_drop
- torchwrench.nn.functional.unsqueeze
- torchwrench.nn.functional.unsqueeze_
- torchwrench.nn.functional.unsqueeze_copy
- torchwrench.nn.functional.view_as_complex
- torchwrench.nn.functional.view_as_real
- torchwrench.nn.functional.where
- torchwrench.nn.functional.zeros
- Submodules
- torchwrench.nn.modules package
- torchwrench.nn.modules.Abs
- torchwrench.nn.modules.Angle
- torchwrench.nn.modules.AsTensor
- torchwrench.nn.modules.CropDim
- torchwrench.nn.modules.CropDims
- torchwrench.nn.modules.EModule
- torchwrench.nn.modules.EModuleDict
- torchwrench.nn.modules.EModuleList
- torchwrench.nn.modules.EModulePartial
- torchwrench.nn.modules.ESequential
- torchwrench.nn.modules.Exp
- torchwrench.nn.modules.Exp2
- torchwrench.nn.modules.FFT
- torchwrench.nn.modules.IFFT
- torchwrench.nn.modules.Identity
- torchwrench.nn.modules.Imag
- torchwrench.nn.modules.IndexToName
- torchwrench.nn.modules.IndexToOnehot
- torchwrench.nn.modules.IndicesToMultihot
- torchwrench.nn.modules.IndicesToMultinames
- torchwrench.nn.modules.Log
- torchwrench.nn.modules.Log10
- torchwrench.nn.modules.Log2
- torchwrench.nn.modules.LogSoftmaxMultidim
- torchwrench.nn.modules.MaskedMean
- torchwrench.nn.modules.MaskedSum
- torchwrench.nn.modules.Max
- torchwrench.nn.modules.Mean
- torchwrench.nn.modules.Min
- torchwrench.nn.modules.Module
- torchwrench.nn.modules.ModuleDict
- torchwrench.nn.modules.ModuleList
- torchwrench.nn.modules.ModulePartial
- torchwrench.nn.modules.MoveToRec
- torchwrench.nn.modules.MultiIndicesToMultihot
- torchwrench.nn.modules.MultiIndicesToMultinames
- torchwrench.nn.modules.MultihotToIndices
- torchwrench.nn.modules.MultihotToMultiIndices
- torchwrench.nn.modules.MultihotToMultinames
- torchwrench.nn.modules.MultilabelToPowerset
- torchwrench.nn.modules.MultinamesToIndices
- torchwrench.nn.modules.MultinamesToMultiIndices
- torchwrench.nn.modules.MultinamesToMultihot
- torchwrench.nn.modules.NDArrayToTensor
- torchwrench.nn.modules.NameToIndex
- torchwrench.nn.modules.NameToOnehot
- torchwrench.nn.modules.Normalize
- torchwrench.nn.modules.OnehotToIndex
- torchwrench.nn.modules.OnehotToName
- torchwrench.nn.modules.PadAndCropDim
- torchwrench.nn.modules.PadAndStackRec
- torchwrench.nn.modules.PadDim
- torchwrench.nn.modules.PadDims
- torchwrench.nn.modules.Permute
- torchwrench.nn.modules.PositionalEncoding
- torchwrench.nn.modules.Pow
- torchwrench.nn.modules.PowersetToMultilabel
- torchwrench.nn.modules.ProbsToIndex
- torchwrench.nn.modules.ProbsToIndices
- torchwrench.nn.modules.ProbsToMultiIndices
- torchwrench.nn.modules.ProbsToMultihot
- torchwrench.nn.modules.ProbsToMultinames
- torchwrench.nn.modules.ProbsToName
- torchwrench.nn.modules.ProbsToOnehot
- torchwrench.nn.modules.Real
- torchwrench.nn.modules.Repeat
- torchwrench.nn.modules.RepeatInterleave
- torchwrench.nn.modules.RepeatInterleaveNd
- torchwrench.nn.modules.ResampleNearestFreqs
- torchwrench.nn.modules.ResampleNearestRates
- torchwrench.nn.modules.ResampleNearestSteps
- torchwrench.nn.modules.Reshape
- torchwrench.nn.modules.Sequential
- torchwrench.nn.modules.Shuffled
- torchwrench.nn.modules.SoftmaxMultidim
- torchwrench.nn.modules.Sort
- torchwrench.nn.modules.Squeeze
- torchwrench.nn.modules.TFlatten
- torchwrench.nn.modules.TensorTo
- torchwrench.nn.modules.TensorToNDArray
- torchwrench.nn.modules.ToItem
- torchwrench.nn.modules.ToList
- torchwrench.nn.modules.ToNDArray
- torchwrench.nn.modules.ToTensor
- torchwrench.nn.modules.TopP
- torchwrench.nn.modules.Topk
- torchwrench.nn.modules.TransformDrop
- torchwrench.nn.modules.Transpose
- torchwrench.nn.modules.Unsqueeze
- torchwrench.nn.modules.View
- torchwrench.nn.modules.ViewAsComplex
- torchwrench.nn.modules.ViewAsReal
- Submodules
- torchwrench.nn.functional package
- torchwrench.optim package
- torchwrench.serialization package
- torchwrench.serialization.as_builtin
- torchwrench.serialization.dump_csv
- torchwrench.serialization.dump_json
- torchwrench.serialization.dump_pickle
- torchwrench.serialization.dump_to
- torchwrench.serialization.dump_torch
- torchwrench.serialization.load_csv
- torchwrench.serialization.load_from
- torchwrench.serialization.load_pickle
- torchwrench.serialization.load_torch
- torchwrench.serialization.read_from
- torchwrench.serialization.save_to
- Submodules
- torchwrench.serialization.common module
- torchwrench.serialization.csv module
- torchwrench.serialization.dump_fn module
- torchwrench.serialization.hdf module
- torchwrench.serialization.json module
- torchwrench.serialization.load_fn module
- torchwrench.serialization.numpy module
- torchwrench.serialization.pickle module
- torchwrench.serialization.safetensors module
- torchwrench.serialization.torch module
- torchwrench.serialization.torchaudio module
- torchwrench.serialization.yaml module
- torchwrench.types package
- torchwrench.types.BoolTensor
- torchwrench.types.BoolTensor0D
- torchwrench.types.BoolTensor1D
- torchwrench.types.BoolTensor2D
- torchwrench.types.BoolTensor3D
- torchwrench.types.ByteTensor
- torchwrench.types.ByteTensor0D
- torchwrench.types.ByteTensor1D
- torchwrench.types.ByteTensor2D
- torchwrench.types.ByteTensor3D
- torchwrench.types.CDoubleTensor
- torchwrench.types.CDoubleTensor0D
- torchwrench.types.CDoubleTensor1D
- torchwrench.types.CDoubleTensor2D
- torchwrench.types.CDoubleTensor3D
- torchwrench.types.CFloatTensor
- torchwrench.types.CFloatTensor0D
- torchwrench.types.CFloatTensor1D
- torchwrench.types.CFloatTensor2D
- torchwrench.types.CFloatTensor3D
- torchwrench.types.CharTensor
- torchwrench.types.CharTensor0D
- torchwrench.types.CharTensor1D
- torchwrench.types.CharTensor2D
- torchwrench.types.CharTensor3D
- torchwrench.types.ComplexFloatingTensor
- torchwrench.types.ComplexFloatingTensor0D
- torchwrench.types.ComplexFloatingTensor1D
- torchwrench.types.ComplexFloatingTensor2D
- torchwrench.types.ComplexFloatingTensor3D
- torchwrench.types.DoubleTensor
- torchwrench.types.DoubleTensor0D
- torchwrench.types.DoubleTensor1D
- torchwrench.types.DoubleTensor2D
- torchwrench.types.DoubleTensor3D
- torchwrench.types.FloatTensor
- torchwrench.types.FloatTensor0D
- torchwrench.types.FloatTensor1D
- torchwrench.types.FloatTensor2D
- torchwrench.types.FloatTensor3D
- torchwrench.types.FloatingTensor
- torchwrench.types.FloatingTensor0D
- torchwrench.types.FloatingTensor1D
- torchwrench.types.FloatingTensor2D
- torchwrench.types.FloatingTensor3D
- torchwrench.types.HalfTensor
- torchwrench.types.HalfTensor0D
- torchwrench.types.HalfTensor1D
- torchwrench.types.HalfTensor2D
- torchwrench.types.HalfTensor3D
- torchwrench.types.IntTensor
- torchwrench.types.IntTensor0D
- torchwrench.types.IntTensor1D
- torchwrench.types.IntTensor2D
- torchwrench.types.IntTensor3D
- torchwrench.types.IntegralTensor
- torchwrench.types.IntegralTensor0D
- torchwrench.types.IntegralTensor1D
- torchwrench.types.IntegralTensor2D
- torchwrench.types.IntegralTensor3D
- torchwrench.types.LongTensor
- torchwrench.types.LongTensor0D
- torchwrench.types.LongTensor1D
- torchwrench.types.LongTensor2D
- torchwrench.types.LongTensor3D
- torchwrench.types.RealTensor
- torchwrench.types.RealTensor0D
- torchwrench.types.RealTensor1D
- torchwrench.types.RealTensor2D
- torchwrench.types.RealTensor3D
- torchwrench.types.ShortTensor
- torchwrench.types.ShortTensor0D
- torchwrench.types.ShortTensor1D
- torchwrench.types.ShortTensor2D
- torchwrench.types.ShortTensor3D
- torchwrench.types.SignedIntegerTensor
- torchwrench.types.SignedIntegerTensor0D
- torchwrench.types.SignedIntegerTensor1D
- torchwrench.types.SignedIntegerTensor2D
- torchwrench.types.SignedIntegerTensor3D
- torchwrench.types.Tensor0D
- torchwrench.types.Tensor1D
- torchwrench.types.Tensor2D
- torchwrench.types.Tensor3D
- torchwrench.types.UnsignedIntegerTensor
- torchwrench.types.UnsignedIntegerTensor0D
- torchwrench.types.UnsignedIntegerTensor1D
- torchwrench.types.UnsignedIntegerTensor2D
- torchwrench.types.UnsignedIntegerTensor3D
- torchwrench.types.is_builtin_number
- torchwrench.types.is_builtin_scalar
- torchwrench.types.is_integral_dtype
- torchwrench.types.is_number_like
- torchwrench.types.is_numpy_number_like
- torchwrench.types.is_numpy_scalar_like
- torchwrench.types.is_scalar_like
- torchwrench.types.is_tensor_or_array
- Submodules
- torchwrench.types.guards module
- torchwrench.types.tensor_subclasses module
- torchwrench.types.tensor_subclasses.BoolTensor
- torchwrench.types.tensor_subclasses.BoolTensor0D
- torchwrench.types.tensor_subclasses.BoolTensor1D
- torchwrench.types.tensor_subclasses.BoolTensor2D
- torchwrench.types.tensor_subclasses.BoolTensor3D
- torchwrench.types.tensor_subclasses.ByteTensor
- torchwrench.types.tensor_subclasses.ByteTensor0D
- torchwrench.types.tensor_subclasses.ByteTensor1D
- torchwrench.types.tensor_subclasses.ByteTensor2D
- torchwrench.types.tensor_subclasses.ByteTensor3D
- torchwrench.types.tensor_subclasses.CDoubleTensor
- torchwrench.types.tensor_subclasses.CDoubleTensor0D
- torchwrench.types.tensor_subclasses.CDoubleTensor1D
- torchwrench.types.tensor_subclasses.CDoubleTensor2D
- torchwrench.types.tensor_subclasses.CDoubleTensor3D
- torchwrench.types.tensor_subclasses.CFloatTensor
- torchwrench.types.tensor_subclasses.CFloatTensor0D
- torchwrench.types.tensor_subclasses.CFloatTensor1D
- torchwrench.types.tensor_subclasses.CFloatTensor2D
- torchwrench.types.tensor_subclasses.CFloatTensor3D
- torchwrench.types.tensor_subclasses.CHalfTensor
- torchwrench.types.tensor_subclasses.CHalfTensor0D
- torchwrench.types.tensor_subclasses.CHalfTensor1D
- torchwrench.types.tensor_subclasses.CHalfTensor2D
- torchwrench.types.tensor_subclasses.CHalfTensor3D
- torchwrench.types.tensor_subclasses.CharTensor
- torchwrench.types.tensor_subclasses.CharTensor0D
- torchwrench.types.tensor_subclasses.CharTensor1D
- torchwrench.types.tensor_subclasses.CharTensor2D
- torchwrench.types.tensor_subclasses.CharTensor3D
- torchwrench.types.tensor_subclasses.ComplexFloatingTensor
- torchwrench.types.tensor_subclasses.ComplexFloatingTensor0D
- torchwrench.types.tensor_subclasses.ComplexFloatingTensor1D
- torchwrench.types.tensor_subclasses.ComplexFloatingTensor2D
- torchwrench.types.tensor_subclasses.ComplexFloatingTensor3D
- torchwrench.types.tensor_subclasses.DoubleTensor
- torchwrench.types.tensor_subclasses.DoubleTensor0D
- torchwrench.types.tensor_subclasses.DoubleTensor1D
- torchwrench.types.tensor_subclasses.DoubleTensor2D
- torchwrench.types.tensor_subclasses.DoubleTensor3D
- torchwrench.types.tensor_subclasses.FloatTensor
- torchwrench.types.tensor_subclasses.FloatTensor0D
- torchwrench.types.tensor_subclasses.FloatTensor1D
- torchwrench.types.tensor_subclasses.FloatTensor2D
- torchwrench.types.tensor_subclasses.FloatTensor3D
- torchwrench.types.tensor_subclasses.FloatingTensor
- torchwrench.types.tensor_subclasses.FloatingTensor0D
- torchwrench.types.tensor_subclasses.FloatingTensor1D
- torchwrench.types.tensor_subclasses.FloatingTensor2D
- torchwrench.types.tensor_subclasses.FloatingTensor3D
- torchwrench.types.tensor_subclasses.HalfTensor
- torchwrench.types.tensor_subclasses.HalfTensor0D
- torchwrench.types.tensor_subclasses.HalfTensor1D
- torchwrench.types.tensor_subclasses.HalfTensor2D
- torchwrench.types.tensor_subclasses.HalfTensor3D
- torchwrench.types.tensor_subclasses.IntTensor
- torchwrench.types.tensor_subclasses.IntTensor0D
- torchwrench.types.tensor_subclasses.IntTensor1D
- torchwrench.types.tensor_subclasses.IntTensor2D
- torchwrench.types.tensor_subclasses.IntTensor3D
- torchwrench.types.tensor_subclasses.IntegralTensor
- torchwrench.types.tensor_subclasses.IntegralTensor0D
- torchwrench.types.tensor_subclasses.IntegralTensor1D
- torchwrench.types.tensor_subclasses.IntegralTensor2D
- torchwrench.types.tensor_subclasses.IntegralTensor3D
- torchwrench.types.tensor_subclasses.LongTensor
- torchwrench.types.tensor_subclasses.LongTensor0D
- torchwrench.types.tensor_subclasses.LongTensor1D
- torchwrench.types.tensor_subclasses.LongTensor2D
- torchwrench.types.tensor_subclasses.LongTensor3D
- torchwrench.types.tensor_subclasses.RealTensor
- torchwrench.types.tensor_subclasses.RealTensor0D
- torchwrench.types.tensor_subclasses.RealTensor1D
- torchwrench.types.tensor_subclasses.RealTensor2D
- torchwrench.types.tensor_subclasses.RealTensor3D
- torchwrench.types.tensor_subclasses.ShortTensor
- torchwrench.types.tensor_subclasses.ShortTensor0D
- torchwrench.types.tensor_subclasses.ShortTensor1D
- torchwrench.types.tensor_subclasses.ShortTensor2D
- torchwrench.types.tensor_subclasses.ShortTensor3D
- torchwrench.types.tensor_subclasses.SignedIntegerTensor
- torchwrench.types.tensor_subclasses.SignedIntegerTensor0D
- torchwrench.types.tensor_subclasses.SignedIntegerTensor1D
- torchwrench.types.tensor_subclasses.SignedIntegerTensor2D
- torchwrench.types.tensor_subclasses.SignedIntegerTensor3D
- torchwrench.types.tensor_subclasses.TTensor
- torchwrench.types.tensor_subclasses.Tensor0D
- torchwrench.types.tensor_subclasses.Tensor1D
- torchwrench.types.tensor_subclasses.Tensor2D
- torchwrench.types.tensor_subclasses.Tensor3D
- torchwrench.types.tensor_subclasses.UnsignedIntegerTensor
- torchwrench.types.tensor_subclasses.UnsignedIntegerTensor0D
- torchwrench.types.tensor_subclasses.UnsignedIntegerTensor1D
- torchwrench.types.tensor_subclasses.UnsignedIntegerTensor2D
- torchwrench.types.tensor_subclasses.UnsignedIntegerTensor3D
- torchwrench.types.variable_fns module
- torchwrench.utils package
- Subpackages
- torchwrench.utils.data package
- torchwrench.utils.data.AdvancedCollateDict
- torchwrench.utils.data.BalancedSampler
- torchwrench.utils.data.CollateDict
- torchwrench.utils.data.SubsetCycleSampler
- torchwrench.utils.data.SubsetSampler
- torchwrench.utils.data.balanced_monolabel_split
- torchwrench.utils.data.get_auto_num_cpus
- torchwrench.utils.data.get_auto_num_gpus
- torchwrench.utils.data.random_split
- Subpackages
- Submodules
- torchwrench.utils.data package
- Submodules
- Subpackages