PyTorch torch.div() method divides every element of the input with a constant and returns a new modified tensor.
Syntax:
torch.div(inp, other, out=None)Arguments
- inp: This is input tensor.
- other: This is a number to be divided to each element of input inp.
- out: The output tensor.
Return: It returns a Tensor.
Let’s see this concept with the help of few examples:
Example 1:
Example 1:
# Importing the PyTorch library import torch # A constant tensor of size n a = torch.FloatTensor([1, 4, 6, 8, 10, 14]) print(a) # Applying the div function and # storing the result in 'out' out = torch.div(a, 0.5) print(out) |
Output:
1 4 6 8 10 14 [torch.FloatTensor of size 6] 2 8 12 16 20 28 [torch.FloatTensor of size 6]
Example 2:
# Importing the PyTorch library import torch # A constant tensor of size n a = torch.randn(6) print(a) # Applying the div function and # storing the result in 'out' out = torch.div(a, 0.3) print(out) |
Output:
-0.8453 -0.1101 0.9431 -0.3041 1.4305 -0.0390 [torch.FloatTensor of size 6] -2.8176 -0.3669 3.1436 -1.0137 4.7683 -0.1300 [torch.FloatTensor of size 6]
<!–
–>
Python | PyTorch sin() method
Python | PyTorch sinh() method
Python | PyTorch cosh() method
Python | PyTorch tanh() method
Python | PyTorch cos() method
Python | PyTorch tan() method
Python | PyTorch asin() method
Python | PyTorch acos() method
Python | PyTorch atan() method
Python PyTorch zeros() method
Please Login to comment…