Tag: GPYTORCH
Estoy tratando de descubrir cómo hacer la siguiente transmisión: Tengo dos tensores, de tamaños (n1,N) y (n2,N) Lo que quiero hacer es multiplicar cada fila del primer tensor con cada fila del segundo tensor, y luego sumar cada resultado de fila multiplicado, de manera que mi tensor final debería tener . . . Read more
Solo sé cómo usar la red incorporada como RNN o LSTM en PyTorch. Pero suelen tratar con cada nodo en la capa anterior que dará información a todos los nodos en la siguiente capa. Quiero hacer algo diferente, pero no sé cómo codificarlo yo mismo. Algo como en esta figura: . . . Read more
Soy un traductor en español que traduce documentos de TI. Traduce lo siguiente al español. No traduzcas el código ni la salida en markdown Soy realmente nuevo en pytorch, y he estado escribiendo el código de convolución yo mismo. Para aplicar convolución en los datos de entrada, uso conv2d. En . . . Read more
En mi caso, estoy aplicando la operación de despliegue en un tensor de A como se muestra a continuación: A.shape = torch.Size([16, 309,128]) A = A.unsqueeze(1) # supongo que es para convertirlo en 4 dimensiones para la operación de despliegue A_out = F.unfold(A, (7, 128), stride=(1, 128), dilation=(3,1)) A_out.shape = . . . Read more
Estoy tratando de liberar memoria de GPU después de terminar de usar el modelo. Verifiqué el estado de la memoria GPU con nvidia-smi antes de crear y entrenar el modelo: 402MiB / 7973MiB Después de crear y entrenar el modelo, verifiqué nuevamente el estado de la memoria GPU con nvidia-smi: . . . Read more