Divide un tensor en tensores `num_split` a lo largo de una dimensión.
Métodos públicos
estática <T> Dividir <T> | |
Iterador < operando <T >> | iterador () |
Lista < salida <T >> | salida () Son tensores de forma idéntica, cuya forma coincide con la de `value` excepto a lo largo de` axis`, donde sus tamaños son `values.shape [split_dim] / num_split`. |
Métodos heredados
Métodos públicos
public static de Split <T> crear ( Alcance alcance, operando <Integer> eje, operando <T> valor, Long numSplit)
Método de fábrica para crear una clase que envuelva una nueva operación de división.
Parámetros
alcance | alcance actual |
---|---|
eje | 0-D. La dimensión a lo largo de la cual dividirse. Debe estar en el rango `[-rango (valor), rango (valor))`. |
valor | El tensor para dividir. |
numSplit | El número de formas de dividir. Debe dividir equitativamente "value.shape [split_dim]". |
Devoluciones
- una nueva instancia de Split
Lista pública < salida <T >> salida ()
Son tensores de forma idéntica, cuya forma coincide con la de `value` excepto a lo largo de` axis`, donde sus tamaños son `values.shape [split_dim] / num_split`.