SplitV public final class
Divide un tensor en tensores `num_split` a lo largo de una dimensión.
Métodos públicos
estática <T, U se extiende Número> SplitV <T> | |
Iterador < operando <T >> | iterador () |
Lista < salida <T >> | salida () Tensores cuya forma coincide con la de `value` excepto a lo largo de` axis`, donde sus tamaños son `size_splits [i]`. |
Métodos heredados
Métodos públicos
public static SplitV <T> crear ( Alcance alcance, operando <T> valor, operando <U> sizeSplits, operando <Integer> eje, Long numSplit)
Método de fábrica para crear una clase que envuelva una nueva operación SplitV.
Parámetros
alcance | alcance actual |
---|---|
valor | El tensor para dividir. |
tamañoSplits | 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 la dimensión se va a inferir. |
eje | 0-D. La dimensión a lo largo de la cual dividirse. Debe estar en el rango `[-rango (valor), rango (valor))`. |
Devoluciones
- una nueva instancia de SplitV
Lista pública < salida <T >> salida ()
Tensores cuya forma coincide con la de `value` excepto a lo largo de` axis`, donde sus tamaños son `size_splits [i]`.