Python – PyTorch div() method
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]