From b570080424e9d8af5bb91a032bca2ac9036fb407 Mon Sep 17 00:00:00 2001 From: Philip May Date: Thu, 19 Sep 2024 09:45:07 +0200 Subject: [PATCH] Update pytorch.md --- source/machine-learning/pytorch.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/source/machine-learning/pytorch.md b/source/machine-learning/pytorch.md index c68c127..df4333a 100644 --- a/source/machine-learning/pytorch.md +++ b/source/machine-learning/pytorch.md @@ -2,4 +2,7 @@ ## Commands -- [torch.Tensor.item()](https://pytorch.org/docs/stable/generated/torch.Tensor.item.html): convert tensor to standard Python number - only works for tensors with one element +- [torch.Tensor.item()](https://pytorch.org/docs/stable/generated/torch.Tensor.item.html) + - convert tensor to standard Python number + - only works for tensors with one element + - implicitly moves the value to the CPU