Что такое ComputeType of torch.nn.functional.lineear, когда вход Float16 или Bfloat16 ⇐ Python

[Anonymous]

В следующем коде < /p>

Код: Выделить всё

import torch
from torch.nn.functional import linear
a=torch.ones(2,3).type(torch.float16)
b=torch.ones(2,3).type(torch.float16)
linear(a,b)

Что такое Computepe линейного, FP32 или FP16 или другого?>

Подробнее здесь: https://stackoverflow.com/questions/795 ... oat16-or-b