Linear(n_feature, n_hidden) self.predict = torch.nn. ... For a dataset of 100 samples, if the batch size is 4 meaning we have 25 batches.
確定! 回上一頁