index_select() in PyTorch
index_select() can get the 0D or more D tensor of the zero or more elements selected with zero or more indices, not removing one dimension from the 0D or more D tensor of zero or more elements as shown below:...
Running PyTorch on GPUs – DZone
Running an AI workload on a GPU machine requires the installation of kernel drivers and user space libraries from GPU vendors such as AMD and NVIDIA. Once the driver and software are installed, to use AI frameworks such as PyTorch...
Set out with out argument functions PyTorch
You can set out with the functions which have out argument as shown below: *Memos: I selected some popular keepdim argument functions such as arange(), rand() add(), mean(), median(), min(), max(), all(), any() and matmul(). out(Optional, tensor) can output a...
Set keepdim with keepdim argument functions PyTorch
You can set keepdim with the functions which have keepdim argument as shown below: *Memos: sum(). *My post explains sum(): import torch my_tensor = torch.tensor([1, 2, 3, 4]) torch.sum(input=my_tensor) torch.sum(input=my_tensor, dim=0) # tensor(10) torch.sum(input=my_tensor, dim=0, keepdim=True) # tensor([10]) prod(). *My...
Set dtype with dtype argument functions and get it in PyTorch
You can set dtype with the functions which have dtype arguments and get it with dtype and type() as shown below: *Memos: tensor(). *My post explains tensor(): import torch my_tensor = torch.tensor([0, 1, 2]) my_tensor = torch.tensor([0, 1, 2], dtype=torch.int64)...
Python Deep Learning: PyTorch vs Tensorflow – Real Python
PyTorch vs TensorFlow: What’s the difference? Both are open source Python libraries that use graphs to perform numerical computation on data. Both are used extensively in academic research and commercial code. Both are extended by a variety of APIs, cloud...
PyTorch vs TensorFlow for Your Python Deep Learning Project – Real Python
PyTorch vs TensorFlow: What’s the difference? Both are open-source Python libraries that use graphs to perform numerical computations on data in deep learning applications. Both are used extensively in academic research and commercial code. Both are extended by a variety...
equal(), eq() and ne() in PyTorch
equal() can check 2 tensors are the same: *Memos: equal() can be called both from torch and a tensor. The tensors can be 0D or more D tensors. import torch tensor1 = torch.tensor([5, 9, 0, 1]) tensor2 = torch.tensor([5, 9,...
min(), max(), aminmax(), amin(), amax(), argmin() and argmax() in PyTorch
min() can get one or more minimum values as shown below: *Memos: min() can be called both from torch and a tensor. Setting a dimension(dim) to the 2nd argument with torch or the 1st argument with a tensor gets zero...
min(), max(), aminmax(), amin(), amax(), argmin() and argmax() in PyTorch
min() can get one or more minimum values as shown below: *Memos: min() can be called both from torch and a tensor. Setting a dimension(dim) to the 2nd argument with torch or the 1st argument with a tensor gets zero...
