preguntar acerca de pytorch

import torch,ipdb import torch.autograd as autograd import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torch.autograd import Variable rnn = nn.LSTM(input_size=10, hidden_size=20, num_layers=2) input = Variable(tor …

par exemple, j'ai un vecteur 1D avec la dimension (5). Je voudrais le remodeler en matrice 2D (1,5). Voici comment je le fais avec numpy >>> import numpy as np >>> a = np.array([1,2,3,4,5]) >>> a.shape (5,) >>> a …

j'essayais de faire une chose simple qui était d'entraîner un modèle linéaire avec une descente en pente Stochastique (SGD) en utilisant la torche: import numpy as np import torch from torch.autograd import Variable import pdb def get_batch2(X,Y, …

comment utiliser le torch.utils.data.Dataset et torch.utils.data.DataLoader sur vos propres données (pas seulement l' torchvision.datasets)? Est-il possible d'utiliser les fonctions intégrées DataLoaders qu'ils utilisent sur TorchVisionDatasets pour …

y a-t-il un moyen, je peux imprimer le résumé d'un modèle à PyTorch comme model.summary() méthode fait en Kéras comme suit? Model Summary: ____________________________________________________________________________________________________ Layer (ty …