我怎样才能制作火炬张量？

Question

我想制作一个仅由 1 或 -1 组成的 torch 张量。

我只是尝试使用 torch.empty() 和 torch.randn()。

tmp = torch.randint(low=0, high=2, size=(4, 4))
tmp[tmp==0] = -1
tmp
>> tensor([[ 1, -1, -1,  1],
           [-1,  1,  1, -1],
           [-1, -1,  1,  1],
           [-1,  1, -1,  1]])

但是，我不知道什么方法在时间上最有效。

而且我想把这段代码尽量写成一行，因为这段代码位于__init__ ()

Answer 1

为什么不呢：

tmp = -1 + 2 * torch.randint(low=0, high=2, size=(4, 4))

我怎样才能制作火炬张量？

How can I make torch tensor?

pytorch

tensor