import  torchfrom    torch import nn, optim, autogradimport  numpy as npimport  visdomfrom    torch.nn import functional as Ffrom    matplotlib import pyplot as pltimport  random

h_dim = 400batchsz = 512viz