下载
中文
注册

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

支持bf16,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