mlr_pipeops_trafo_reshape {mlr3torch}R Documentation

PipeOpPreprocTorchTrafoReshape

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


[Package mlr3torch version 0.1.1 Index]