Function create_torch_sequential

tests/test_tensor_broadcast.cpp:90–102

Helper to create PyTorch sequential tensor

    // Helper to create PyTorch sequential tensor
    torch::Tensor create_torch_sequential(const std::vector<int64_t>& shape,
                                          torch::Device device = torch::kCUDA) {
        int64_t n = 1;
        for (auto s : shape)
            n *= s;

        std::vector<float> data(n);
        for (int64_t i = 0; i < n; ++i) {
            data[i] = static_cast<float>(i);
        }

        return create_torch_tensor(data, shape, device);
    }

} // anonymous namespace