SplitV

clase final pública SplitV

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

Métodos públicos

estático <T, U extiende Número> SplitV <T>
crear (alcance alcance , valor del operando <T>, divisiones de tamaño del operando <U>, eje del operando <entero>, división numérica larga)
Método de fábrica para crear una clase que envuelve una nueva operación SplitV.
Iterador< Operando <T>>
iterador ()
Lista< Salida <T>>
producción ()
Tensores cuya forma coincide con la del "valor" excepto a lo largo del "eje", donde sus tamaños son "size_splits[i]".

Métodos heredados

Métodos públicos

creación pública estática de SplitV <T> (alcance del alcance , valor del operando <T>, división de tamaño del operando <U>, eje del operando <entero>, división numérica larga)

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

Parámetros
alcance alcance actual
valor El tensor a dividir.
tamañoDivisiones lista que contiene los tamaños de cada tensor de salida a lo largo de la dimensión dividida. Debe sumar la dimensión del valor a lo largo de split_dim. Puede contener un -1 que indica que se va a inferir la dimensión.
eje 0-D. La dimensión a lo largo de la cual dividirse. Debe estar en el rango `[-rank(valor), rango(valor))`.
Devoluciones
  • una nueva instancia de SplitV

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

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

Tensores cuya forma coincide con la del "valor" excepto a lo largo del "eje", donde sus tamaños son "size_splits[i]".