trw.simple_layers.sub_tensor

Module Contents

Classes

SubTensor

Select a region of a tensor (without copy), excluded the first component (N)

class trw.simple_layers.sub_tensor.SubTensor(node, min_indices, max_indices_exclusive)

Bases: trw.simple_layers.simple_layers.SimpleModule

Select a region of a tensor (without copy), excluded the first component (N)