小编mat*_*dns的帖子

有一个功能np.split()可以沿1轴分割数组.我想知道是否有一个多轴版本,你可以沿轴(0,1,2)分割.

我试图以两种方式重复火炬的张量.例如,{1,2,3,4}两种方式重复张量3次以产生;

{1,2,3,4,1,2,3,4,1,2,3,4}
{1,1,1,2,2,2,3,3,3,4,4,4}

有一个内置的火炬:repeatTensor函数,它将生成两个中的第一个(如numpy.tile()),但我找不到后者(如numpy.repeat()).我确信我可以在第一次调用sort来给第二个,但我认为对于更大的数组,这可能在计算上是昂贵的？

谢谢.

当我跑;

require 'cutorch'

在lua中,它会自动为GPU中的两个核心分配两个进程.例如,我得到以下输出nvidia-smi;

---------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|    0                  Not Supported                                         |
|    1      6091    C   /home/msmith/torch/install/bin/qlua             98MiB |
|    2      6091    C   /home/msmith/torch/install/bin/qlua             99MiB |
+-----------------------------------------------------------------------------+

我希望能够控制进程中的GPU.我试过了;

cutorch.setDevice(<Device Number>)

但这只会在GPU上创建更多进程.

谢谢.