torch.utils.tensorboard

若API“是否支持”“是”“限制与说明”“-”,说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.utils.tensorboard.writer.SummaryWriter

-

torch.utils.tensorboard.writer.SummaryWriter.__init__

-

torch.utils.tensorboard.writer.SummaryWriter.add_scalar

-

torch.utils.tensorboard.writer.SummaryWriter.add_scalars

-

torch.utils.tensorboard.writer.SummaryWriter.add_histogram

-

torch.utils.tensorboard.writer.SummaryWriter.add_image

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_images

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_figure

-

torch.utils.tensorboard.writer.SummaryWriter.add_video

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_audio

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_text

-

torch.utils.tensorboard.writer.SummaryWriter.add_graph

支持bf16,fp16,fp32

torch.utils.tensorboard.writer.SummaryWriter.add_embedding

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_pr_curve

支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_custom_scalars

-

torch.utils.tensorboard.writer.SummaryWriter.add_mesh

支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_hparams

-

torch.utils.tensorboard.writer.SummaryWriter.flush

-

torch.utils.tensorboard.writer.SummaryWriter.close

-