Exemplo n.º 1
0
        static TorchTensor Batchify(TorchTensor data, int batch_size)
        {
            var nbatch = data.shape[0] / batch_size;
            var d2     = data.narrow(0, 0, nbatch * batch_size).view(batch_size, -1).t();

            return(d2.contiguous());
        }