Split

clase final pública Split

Divide un tensor en tensores `num_split` a lo largo de una dimensión.

Métodos públicos

estático <T> Dividir <T>
crear (alcance del alcance , eje del operando <entero>, valor del operando <T>, división numérica larga)
Método de fábrica para crear una clase que envuelve una nueva operación Split.
Iterador< Operando <T>>
Lista< Salida <T>>
producción ()
Son tensores de forma idéntica, cuya forma coincide con la del "valor" excepto a lo largo del "eje", donde sus tamaños son "valores.forma[split_dim] / num_split".

Métodos heredados

Métodos públicos

división estática pública <T> crear (alcance del alcance , eje del operando <entero>, valor del operando <T>, división numérica larga)

Método de fábrica para crear una clase que envuelve una nueva operación Split.

Parámetros
alcance alcance actual
eje 0-D. La dimensión a lo largo de la cual dividirse. Debe estar en el rango `[-rank(valor), rango(valor))`.
valor El tensor a dividir.
númeroDividir El número de formas de dividir. Debe dividir uniformemente `value.shape[split_dim]`.
Devoluciones
  • una nueva instancia de Split

Iterador público< Operando <T>> iterador ()

Lista pública< Salida <T>> salida ()

Son tensores de forma idéntica, cuya forma coincide con la del "valor" excepto a lo largo del "eje", donde sus tamaños son "valores.forma[split_dim] / num_split".