如何创建给定形状和间隔的张量？

Question

我正在使用 Pytorch，想要在区间 [3,7) 内创建一个形状为 5X3 的随机张量

torch.rand(5,3) 将返回形状为 5 X 3 的随机张量，但是，我无法设置给定的间隔。

请指导。

Answer 1

您可以使用

u ~ U(0, 1)

将

U ~ [a, b]

映射到

u -> (b - a)*u + a

:

(b - a)*torch.rand(5, 3) + a

Answer 2

定义最小值和最大值并使用以下代码：

import torch
max = 7
min = 3
rand_tensor = (max-min)*torch.rand((5, 3)) + min