Tenseurs partiellement empilables

J'ai une sortie d'une couche convolutive dans PyTorch de la forme [b,c,h]où bse trouvent mes lots, cmes canaux et hles fonctionnalités. Je veux les empiler pour les alimenter dans une couche entièrement connectée sans changer les lots, afin qu'ils aient la forme [b, c*h]. Comment puis-je faire ceci?


Solution du problème

Il semble qu'un simple remodelage ou une vue devrait fonctionner :

input.shape # [b,c,h]
reshaped = input.view(input.shape[0],-1) # [b,c*h]

Commentaires

Enregistrer un commentaire

Posts les plus consultés de ce blog

Comment signer la connexion OKEx API version 5 avec websockets ?

Bien que cette question ait reçu une réponse pour les versions précédentes de l'API OKEx utilisant REST, elle ne l'a pas été pour la dernière version 5 de l'API utilisant des websockets. Les docs sont ici. Je reçois l'erreur suivante {"event":"error","msg":"Invalid sign","code":"60007"} , il doit donc y avoir un problème avec l'algorithme de chaîne de signature, mais je n'arrive pas à identifier où je fais une erreur. import hmac import json import time import hashlib import asyncio import websockets passphrase = "XXXX" secret_key = b"XXXX" api_key = "XXXX" timestamp = int(time.time()) print("timestamp: " + str(timestamp)) sign = str(timestamp) + 'GET' + '/users/self/verify' total_params = bytes(sign, encoding= 'utf-8') signature = hmac.new(bytes(secret_key, encoding= 'utf-8'), total_params, digestmod=hashlib.sha256).digest() sign...
Lire la suite

Comment convertir une interface{} qui est vraiment une tranche de types dont le genre est reflect.Int32 en tranche de int32 ?

J'ai ce qui suit: type Int32A int32 type Int32B int32 et souhaite implémenter une fonction qui peut accepter n'importe quelle tranche de types dont le type est reflect.Int32 et le convertir en []int32 . Par example: func ConvertTypeSliceToInt32Slice(es «es-type») []int32 { result:= make([]int32, len(es)) for i:= 0; i < len(result); i++ { result[i] = es[i].(int32) } return result } func caller() { Int32as:= Int32A{1, 2} Int32bs:= Int32B{3, 5} int32as:= ConvertTypeSliceToInt32Slice(Int32as) int32bs:= ConvertTypeSliceToInt32Slice(Int32bs) } Comment cela peut-il être fait avec n'importe quelle définition de type arbitraire dont le genre est reflect.Int32 ? (Contexte : cette fonction sera utilisée pour convertir des tranches de proto enum s ; c'est-à-dire que l'ensemble complet des types est inconnu et illimité, il switch n'est donc pas possible d'effectuer un a sur chaque type). De plus, j'utilise 1.17 donc je ne peux pas utiliser de types pa...
Lire la suite

Créer un fichier CSV à partir d'un fichier texte avec des jetons d'en-tête à l'aide de scripts shell

Je souhaite créer un fichier CSV à partir d'un ensemble de fichiers texte dans un répertoire avec la structure suivante pour les importer ultérieurement dans une base de données. Title: Article title Word Count: 100 Summary: Article summary. Can consist of multiple lines. Keywords: keyword1, keyword2, keyword3 Article Body: The rest of the article body. Till the end of the file. Le résultat souhaité est donc de les avoir dans un fichier CSV avec des sections comme en-têtes avec leur contenu comme suit. Title | Word Count | Summary | Keywords | Article Body | Article title | 100 | Article summary.\nCan... | keyword1, keyword2, keyword3 |... | Article2 title | 110 | Article summary.\nCan... | keyword1, keyword2, keyword3 |... | J'ai essayé quelques approches avec des scripts awk et shell, mais je n'ai pas réussi jusqu'à présent. Des idées? Solution du problème Selon la documentation de COPY, PostgreSQL prend entièrement en charge le format CSV, ainsi qu'un for...
Lire la suite