less than 1 minute read

Torch Basic

Elements




Elements


Random

torch.rand(num)

x = 10 * torch.rand(num)
  • torch.rand will generate the random numbers between 0.0 to 10.0
  • x will have 10 random vales, each between 0.0 to 10.0


Min and Max

torch.min(x)
torch.max(x)

This function will return the maximum or minimum value


Mathematics


Mul

torch.mul(a, b) # a * b. multiple each elements a and b
torch.matmul(a, b) # a @ b. multiple matrix


E.g.

loss[i] = torch.sum(torch.mul(e, e)) / len(x)

loss function.

  • e: error
  • torch.mul(e, e): e error square. same as e.pow(2) or e**e
  • torch.sum: sum of all the error square
  • / len = Divide the sum of squared errors by the number of elements.


argmin

torch.argmin(loss)

return the index of the value of minimize

Tags:

Categories:

Updated:

Leave a comment