Question

我有

x = torch.ones(n,3) 
x[:,0].uniform_(-1.,1)
x[:5]

我知道了

tensor([[-0.4899,  1.0000,  1.0000],
        [ 0.7395,  1.0000,  1.0000],
        [-0.3599,  1.0000,  1.0000],
        [ 0.0863,  1.0000,  1.0000],
        [ 0.8935,  1.0000,  1.0000]])

我要实现的是将随机值分配给第一行和第二行并使第一行平方

[0.25,0.5,1]（示例）

我有几个问题：

如何将相同的uniform_仅应用于两行？
是否存在平方函数，以及如何将其仅应用于第一行？
可以一次完成吗？

Answer 1

我能够做到这一点

x = torch.ones(n,3) 
x[:,1].uniform_(-1.,1.)
x[:,0] = x[:,1]**2
x[:5]

有输出

tensor([[ 0.8429, -0.9181,  1.0000],
        [ 0.0638,  0.2525,  1.0000],
        [ 0.5168,  0.7189,  1.0000],
        [ 0.7832,  0.8850,  1.0000],
        [ 0.2592,  0.5091,  1.0000]])

我仍然很乐意接受更好的答案：）

如何在Python中的Uniform_ Form Torch之后应用操作？

1 个答案: