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

Question

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

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

请指导。

Answer 1

您可以使用 U ~ [0, 1] 将 U ~ [a, b] 映射到 u -> (a - b)*u + b：

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

Answer 2

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

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