给定两个各含 N 个 float 的数组 a 和 b,把它们逐元素相加,结果写进 c:
c[i] = a[i] + b[i]
solve,签名保持不变。
extern "C" void solve(const float* a, const float* b, float* c, int N)def solve(a: torch.Tensor, b: torch.Tensor, c: torch.Tensor, N: int)a、b、c 已经在 GPU 上。不要修改 a 和 b。c;进入 solve 时 c 里的内容是未定义的。在 Python 里写 c = a + b
只是把局部变量名指向新张量,c 本身一个元素也没写。输入:a = [1.0, 2.0, 3.0, 4.0], b = [10.0, 20.0, 30.0, 40.0], N = 4
输出:c = [11.0, 22.0, 33.0, 44.0]
输入:a = [0.5, -1.5, 2.0], b = [-0.5, 1.0, 0.25], N = 3
输出:c = [0.0, -0.5, 2.25]
N = 25,000,000TODO
CUDA · T4
载入中…
CUDA · T4
载入中…
CUDA · T4
载入中…