Pytorch es un marco de código abierto disponible con un lenguaje de programación de Python. Tensor es una matriz multidimensional que se utiliza para almacenar los datos. Para usar un tensor, tenemos que importar el módulo de antorcha. Para crear un tensor, el método utilizado es tensor ().
Sintaxis:
antorcha.Tensor (datos)Donde los datos son una matriz multidimensional.
Antorcha.rsqrt ()
Sqrt () en pytorch devuelve la raíz cuadrada recíproca de cada elemento en el tensor de pytorch. Se necesita un parámetro.
Sintaxis:
antorcha.RSQRT (tensor_object)Parámetro:
tensor_object es un tensor
Ejemplo 1:
En este ejemplo, crearemos un tensor con una dimensión que tenga 5 elementos y devuelva las raíces cuadradas recíprocas de estos 5 elementos en un tensor.
#módulo de antorcha de ImportProducción:
Tensor real:Laboral:
Ejemplo 2:
En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en cada fila y devolveremos la raíz cuadrada recíproca de los elementos.
#módulo de antorcha de ImportProducción:
Tensor real:Laboral:
Trabajar con CPU
Si desea ejecutar una función rsqrt () en la CPU, tenemos que crear un tensor con una función CPU (). Esto se ejecutará en una máquina CPU.
Cuando creamos un tensor, esta vez, podemos usar la función CPU ().
Sintaxis:
antorcha.Tensor (datos).UPC()Ejemplo 1:
En este ejemplo, crearemos un tensor con una dimensión que tenga 5 elementos en la CPU y devolveremos las raíces cuadradas recíprocas de estos 5 elementos en un tensor.
#módulo de antorcha de ImportProducción:
Tensor real:Laboral:
Ejemplo 2:
En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en la CPU en cada fila y devolveremos la raíz cuadrada recíproca de los elementos.
#módulo de antorcha de ImportProducción:
Tensor real:Laboral:
Conclusión
En esta lección de Pytorch, discutimos sobre la función rsqrt (). Devuelve la raíz cuadrada recíproca de cada elemento en el tensor Pytorch. Discutimos los dos ejemplos con los diferentes tensores dimensionales para realizar la función rsqrt ().