Функция neg

Функция neg применяется к тензору поэлементно и возвращает новый тензор, содержащий отрицательные значения каждого элемента исходного тензора. Функция принимает один обязательный параметр - тензор. Также доступен альтернативный синтаксис через унарный оператор -.

Синтаксис

torch.neg(tensor)

Пример

Давайте создадим тензор из чисел и применим к нему функцию neg:

import torch

t = torch.tensor([1, -2, 3, -4, 5])
res = torch.neg(t)
print(res)

Результат выполнения кода:

tensor([-1, 2, -3, 4, -5])

Пример

Функцию можно вызывать и как метод тензора:

import torch

t = torch.tensor([1, 2, 3, 4, 5])
res = t.neg()
print(res)

Результат выполнения кода:

tensor([-1, -2, -3, -4, -5])

Пример

Также для отрицания тензора можно использовать унарный оператор -:

import torch

t = torch.tensor([1, 2, 3, 4, 5])
res = -t
print(res)

Результат выполнения кода:

tensor([-1, -2, -3, -4, -5])

Пример

Функция neg работает с тензорами любой размерности:

import torch

t = torch.tensor([
	[1, 2, 3],
	[4, 5, 6],
])
res = torch.neg(t)
print(res)

Результат выполнения кода:

tensor([
	[-1, -2, -3],
	[-4, -5, -6],
])

Смотрите также

функцию abs,
которая возвращает абсолютные значения элементов тензора
функцию sign,
которая возвращает знак каждого элемента тензора
функцию reciprocal,
которая вычисляет обратные значения элементов тензора
функцию logical_not,
которая применяет логическое отрицание к элементам тензора