Base Tensor
Base Tensor Class¤
All representations inherit from BaseTensor.
neojax.tensor.base_tensor.BaseTensor
¤
Abstract base tensor.
All neojax tensors subclass BaseTensor.
Custom tensor classes should inherit from BaseTensor.
to_dense() -> Complex[Array, 'n_corners ...']
¤
Converts tensors to a dense tensor representation.
Returns:
| Type | Description |
|---|---|
Complex[Array, 'n_corners ...']
|
Dense tensors stacked along first axis. |
__call__(corner_idx: int, x_ft_slice: Complex[Array, 'in_c ...']) -> Complex[Array, 'out_c ...']
¤
Contracts the given input slice with the tensor weights for the given corner.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
corner_idx
|
int
|
Corner index of the n-dim FFT hypercube. |
required |
x_ft_slice
|
Complex[Array, 'in_c ...']
|
Mode slice of fourier transformed input. |
required |
Returns:
| Type | Description |
|---|---|
Complex[Array, 'out_c ...']
|
Tensor contraction of weight and input. |