Searched defs:dims_to_flatten (Results 1 – 2 of 2) sorted by relevance
73 …dices_by_dims(const Tensor& indices, const IntArrayRef& sizes, const IntArrayRef& dims_to_flatten){ in flatten_indices_by_dims()
455 def __init__(self, start, end, flattened, dims_to_flatten): argument