PipeOpPreprocTorchTrafoReshape: Reshaping Transformation

mlr_pipeops_trafo_reshapeR Documentation

Reshaping Transformation

Description

Reshapes the tensor according to the parameter shape, by calling torch_reshape(). This preprocessing function is applied batch-wise.

Format

R6Class inheriting from PipeOpTaskPreprocTorch.

Parameters

  • shape :: integer()
    The desired output shape. The first dimension is the batch dimension and should usually be -1.


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.