torch_npu.contrib
torch_npu.contrib.npu_fused_attention_with_layernorm
torch_npu.contrib.npu_fused_attention
torch_npu.contrib.Prefetcher
torch_npu.contrib.DCNv2
torch_npu.contrib.BiLSTM
torch_npu.contrib.Swish
torch_npu.contrib.NpuFairseqDropout
torch_npu.contrib.npu_giou
torch_npu.contrib.npu_ptiou
torch_npu.contrib.npu_iou
def fuse_add_softmax_dropout()
def npu_diou()
def npu_ciou()
class NpuCachedDropout()
class MultiheadAttention()
def npu_single_level_responsible_flags()
def npu_bbox_coder_encode_xyxy2xywh()
def npu_fast_condition_index_put()
class matmul_transpose()
def npu_multiclass_nms()
def npu_batched_multiclass_nms()
class roll()
class Mish()
class SiLU()
class ChannelShuffle()
class LabelSmoothingCrossEntropy()
class ModulatedDeformConv()
class NpuDropPath()
class Focus()
class PSROIPool()
class ROIAlign()
class FusedColorJitter()
def npu_bbox_coder_decode_xywh2xyxy()
def npu_bbox_coder_encode_yolo()
class DropoutWithByteMask()
def dropout_with_byte_mask()
父主题：
PyTorch2.1