torch.utils.tensorboard

若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,int8,uint8,int16,int32,int64,bool

torch.utils.tensorboard.writer.SummaryWriter.add_images

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

torch.utils.tensorboard.writer.SummaryWriter.add_figure

  

torch.utils.tensorboard.writer.SummaryWriter.add_video

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

torch.utils.tensorboard.writer.SummaryWriter.add_audio

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

torch.utils.tensorboard.writer.SummaryWriter.add_text

  

torch.utils.tensorboard.writer.SummaryWriter.add_graph

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

torch.utils.tensorboard.writer.SummaryWriter.add_embedding

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

torch.utils.tensorboard.writer.SummaryWriter.add_pr_curve

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

torch.utils.tensorboard.writer.SummaryWriter.add_custom_scalars

  

torch.utils.tensorboard.writer.SummaryWriter.add_mesh

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

torch.utils.tensorboard.writer.SummaryWriter.add_hparams

  

torch.utils.tensorboard.writer.SummaryWriter.flush

  

torch.utils.tensorboard.writer.SummaryWriter.close