tensorboard_embedding】的更多相关文章

from tensorboardX import SummaryWriter import torchvision writer=SummaryWriter(log_dir="embedding") mnist=torchvision.datasets.MNIST("mnist",download=True) writer.add_embedding( mat=mnist.train_data.reshape((-1,28*28))[:100,:], metadat…