La Data Science

Tout sur la data

Calculer le Maximum Entre Deux Tenseurs PyTorch

Problème :

Comment calculer le maximum entre deux tenseurs PyTorch?

Solution :

Utiliser la fonction maximum du module torch.

Exemple :

	import torch

	max_tensor = torch.maximum(x1, x2) # où x1 et x2 sont deux tenseurs que vous avez définis au préalable.

view raw maximum-torch.py hosted with ❤ by GitHub

Jean MP