Add files using upload-large-folder tool

d91766b verified 3 months ago

208 Bytes

from __future__ import annotations

from dataclasses import dataclass

import torch


@dataclass(frozen=True)
class TopKOutput:
    weights: torch.Tensor
    ids: torch.Tensor
    router_logits: torch.Tensor

	from __future__ import annotations

	from dataclasses import dataclass

	import torch


	@dataclass(frozen=True)
	class TopKOutput:
	weights: torch.Tensor
	ids: torch.Tensor
	router_logits: torch.Tensor